anaconda 安装pytorch（GPU版本）教程

最近按照 anaconda的pytorch一直出错，不是版本没对应上就是网络延迟一直下不好

刚看了好几个教程  现在 终于安装好了！希望能够帮到大家

首先是在anaconda prompt 里面创建一个虚拟环境

conda create -n pytorch python=3.8

根据自己的Python版本，可以低于自己的版本，主要还是看自己的cuda，我安装到是3.8

进入虚拟环境：

conda activate pytorch

接下来就是选择适合自己电脑的pytorch torchvision torchaudio cudatoolkit的版本，这个特别重要，因为这个导致重新下载很多回。查看自己显卡的cuda，我的是11.0，向下兼容就行

然后再选择合适的  一定是对应好的版本：

 

 

 最后我选择了这个：

conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=10.2 -c pytorch

然后安装即可

！！！安装如果特别慢就更改镜像源，例如清华源，这里就不在 赘述，网上都有教程，动动手指，切记一定要关上外网服务器，不然不会下载资源。最后等待下载，大致10-30mins。

然后检查一下是否为GPU版本的：

import  torch
torch.cuda.is_available()
print(torch.cuda.get_device_name(0))

显示：

至此设置成功！