GPU版本pytorch（Cuda12.1）清华源快速安装一步一步教！小白教学

见贤思齐

GPU版本pytorch（Cuda12.1）清华源快速安装一步一步教！作者为新电脑r9000p装pytorch时，没有注意自己的cuda版本是12.1（小伙伴自行百度下如何查看），装了旧版的torch结果import报错！找了半天原因才发现跟cuda版本不对应，要下载Cuda11.8这个版本才可以，如下图：pip3installtorchtorchvisiontorchaudio--index-urlhttps://download.pytorch.org/whl/cu1181上面是官方链接，不知道为什么我科学上网了下的还是非常慢，而且看网上有下好之后还有问题的，于是果断选择清华源！下面是一步一步教程：首先确保安装了anaconda，在anacondaprompt命令行窗口创建虚拟环境，我命名为pytorch310（Python版本不要低于3.8）condacreate-npytorch310python==3.101然后激活它：condaactivatepytorch3101设置清华源，加速安装：pipconfigsetglobal.index-urlhttps://pypi.tuna.tsinghua.edu.cn/simple1开始指定版本安装：pipinstalltorch==2.0.0+cu118torchvision==0.15.1+cu118torchaudio==2.0.1+cu118-fhttps://download.pytorch.org/whl/torch_stable.html1应该会非常快！我的下载速度平均在40M/s，两分钟就下好了！然后输入condalist看一下有哪些库，这三个出来就没问题了！不太放心，再试一下：输入python，然后importtorch，这个时候稍微等一下有点慢，然后torch.cuda.is_available()，如下图所示，出来True说明可以调用cuda！gpu版本pytorch安装成功！