PyTorch 安装及常见问题

PyTorch 官网：https://pytorch.org/

一、环境检查

安装前，首先确定已经在相应的平台上已经搭建好了深度学习环境。

• Windows: https://blog.csdn.net/qq_40750972/article/details/115534242
• Linux: https://blog.csdn.net/qq_40750972/article/details/123835847

详细安装参考官网给出的文档：

• https://docs.nvidia.com/
• https://docs.nvidia.com/cuda/index.html
• https://docs.nvidia.com/deeplearning/cudnn/index.html

二、安装 PyTorch

无论是使用 conda 还是 pip 安装时，最好直接使用官方给的安装命令，因为一些版本只能使用 conda install，一些只能使用 pip install！

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

• 这里的 cudatoolkit 版本必须和 nvcc -V 中对应 ，如果不匹配，可能会自动安装 CPU 版本！
• 官方提供的安装命令去掉最后的 -c ...，这是指定下载的通道，优先级比我们自己配置在 .condarc 中的高！
• 注意：如果使用 conda install 无法安装，且确定清华源已经配置好，同时代理已经关闭，还是不行，可以换种方式进行安装，即 pip install，只是需要注意版本对应关系。注意这里不要把最后的 -f 去掉，不然会提示找不到下载源！
  

使用 pip install 安装（同样要和我们的机器对应的 CUDA 版本相匹配 ），一般不用安装最新的版本，版本往前推个 2~3 的比较稳定。

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html