原创博客，转载请注明出处，谢谢！

本文主要介绍 cuda，pytorch，torchvision 在安装过程中的步骤和注意事项。

修改软件源 (只是建议)

原因：用 pip 和 conda 命令安装的时候经常失败，常常需要去找离线的安装包，不方便而且下载速度很慢。

如果你没有以上问题，该步跳过。

解决方案：改成国内的源就可以

打开 CMD 或者 Anaconda Prompt 输入修改指令：

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

或者：

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

查版本

首先确认显卡驱动已经安装，没安装的先安装。

如果只安装 CPU 版本，跳过 GPU 相关内容。

确定 Python 版本

python -V

查看 GPU 版本

win7: “此电脑” 右键 “属性” → “设备管理器” → “显示适配器”，查看版本信息。

win10: “计算机” 右键 “属性” → 相关设置 “设备管理器” → “显示适配器”，查看版本信息。

各包对应版本

pytorch torchvision python cuda
1.8.0 0.9.0 >=3.6 10.2, 11.1
1.7.1 0.8.2 >=3.6 9.2, 10.1, 10.2, 11.0
1.7.0 0.8.0 >=3.6 9.2, 10.1, 10.2, 11.0
1.6.0 0.7.0 >=3.6 9.2, 10.1, 10.2
1.5.1 0.6.1 >=3.6 9.2, 10.1, 10.2
1.5.0 0.6.0 >=3.6 9.2, 10.1, 10.2
1.4.0 0.5.0 2.7, >=3.5, <=3.8 9.0, 10.0
1.3.1 0.4.2 2.7, >=3.5, <=3.7 9.0, 10.0
1.3.0 0.4.1 2.7, >=3.5, <=3.7 9.0, 10.0
1.2.0 0.4.0 2.7, >=3.5, <=3.7 9.0, 10.0
1.1.0 0.3.0 2.7, >=3.5, <=3.7 9.0, 10.0
<1.0.1 0.2.2 2.7, >=3.5, <=3.7 9.0, 10.0

安装

根据官方网页的配置设定生成命令：

官方网页：Pytorch

这些配置常常不是我们需要的，就查看历史版本里的命令配置：Previous vsesions of Pytorch

但是，更多时候还是需要自己一个一个包手动安装。

离线包下载

先尝试在线 pip 安装每一个包，如果不成功，去相应网站根据你的配置选择对应的包后安装。

下载地址：pytorch 下载

如果速度慢也可以上清华源，但是清华源有些版本没有。