前提:需要安装完Anaconda3

详细请参考此链接:
https://www.bilibili.com/read/cv6124348

第一步

打开开始菜单,直接输入anaconda,打开 anaconda prompt

结果:

第二步

我们需要更换清华大学开源下载库(直接从官网下载容易下载失败):

1
2
3
4
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels

第三步

替换完下载源后我们就可以开始安装啦:

1
conda install pytorch torchvision cudatoolkit=10.1

后面只要跟着提示走就可以了,作者因为已经安装完成所以这里不再做演示
若下载中出现网络连接失败情况,请重新尝试几次
老版本的pytorch可以进行手动删除或者不删,因为高版本会替换掉低版本

第四步

安装完后,我们可以测试一下,可以直接在 anaconda prompt里测试,也可以使用Jupyter, Spyder中进行测试

Eg.

如果你的结果和我上面这张图一样,那么就恭喜你!安装成功啦!