PyTorch最新安装教程——torch1.6 + torchvision0.7（2022.04)

2022.4.18更新

安装流程

1、安装 Anaconda3 2、检查显卡，更新驱动 3、创建PyTorch环境 4、安装 PyTorch（重点） 5、测试

经过N次的安装，每次都要翻大佬们的blog，痛并思痛，还是记录下安装历程吧!

1、安装 Anaconda Anaconda 是一个用于科学计算的 Python 发行版，支持 Linux, Mac, Windows, 包含了众多流行的科学计算、数据分析的 Python 包。

Anaconda安装包下载： 用的是清华源的下载，也可以去官网下载

2、检查显卡，更新驱动（电脑较新的可以跳过这步）

首先，建议到英伟达官网更新驱动，按照最新的来安装（这里有人会说不更新也行，但更新最新的驱动是为了避免不必要的麻烦）： 选择自己的显卡型号，下载安装即可 然后，安装好后进入NVIDIA控制面板（一般电脑右下角就会看到英伟达的图标），鼠标右键可进入。点击帮助，选择系统信息。

点击组件，找到自己CUDA的版本号 也可以 win +R ->cmd ，进入命令行，输入：nvidia-smi ，即可获得cuda版本号

3、 创建PyTorch环境

进入 Anaconda prompt 命令窗口 创建虚拟环境 输入命令：conda create -n PyTorch python=3.8 其中，PyTorch是虚拟环境的名字，自己随意设置；3.8是python版本，可以自己选择安装3.7或3.6版本（不建议安装3.9），但一定要指定python版本；

创建成功后，查看刚才创建的环境，输入以下命令： conda info --envs

激活环境： conda activate PyTorch

4、安装PyTorch

进入PyTorch官网：

点击Install，官网会自动根据你的电脑，显示的即是你可安装的CUDA版本，并给出安装命令。

复制官网提供的命令： conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch 最新更新： 以上给出的命令是基于CUDA11.3的，但想要安装CUDA10.2的版本，官网不再提供安装命令（但可以这个链接找到安装命令），亲测并不是很好用，很大机率安装失败，建议使用以下方法： 首先，到这下载自己需要的 torch 和 torchvision 版本 ： cu102/torch-1.6.0-cp37-cp37m-win_amd64.whl cu102/torchvision-0.7.0-cp37-cp37m-win_amd64.whl 其中，cu102就代表cuda10.2

然后激活进入PyTorch环境： 再找到自己刚下载的whl文件路径，cd 进入到路径下，输入以下命令：

pip install torch-1.6.0-cp37-cp37m-win_amd64.whl
pip install torchvision-0.7.0-cp37-cp37m-win_amd64.whl

5、测试

打开 Anaconda prompt 命令窗口 进入PyTorch环境：conda activate PyTorch 输入python 进入python开发环境中 输入以下命令： import torch torch.cuda.is_available() 显示True，则安装成功，至此安装完成。

附录

如在pycharm中遇到虚拟环境问题，可参考： Pycharm加载conda创建pytorch虚拟环境 & import torch报错问题解决 链接：