Linux下从0开始GPU环境搭建与启动测试

文章正文
发布时间:2024-08-23 10:39

torch-torchvision-cuda版本关系对应 CUDA Toolkit 和Pytorch的对应关系 说明: 用anaconda安装torch。新建虚拟环境后,直接在pytorch官网官网链接找到“Install”按钮。这里一键搞定torch,torchvision,cudatoolkit等等,不需要另外安装cuda(笔者在没有单独安装CUDA情况下,成功运行了torch-gpu,很丝滑),并且版本都会自己对于对应好,原因是anaconda都集成在虚拟环境里面了,统一管理各个依赖包。