Anaconda Pytorch 安装方法

前言

有一段时间比较喜欢用pytorch，主要是因为pytorch各个版本的安装命令写的很清楚，配合anaconda使用也不需要特别关心电脑当前装的显卡驱动版本、cuda版本等（保持最新就行）

但是偶尔会遇到

import torch
print(torch.__version__)
print(torch.cuda.is_available())

提示False的问题

查了资料后发现是conda的问题，虽然安装命令指定要装GPU版本，但是conda在源中没有找到完全匹配的python、pytorch、cudatoolkit、cudnn包，就会找个CPU版本的包安装

解决方法

1. 在清华源(等)镜像网站，查询python与pytorch对应版本

Windows 64位清华源地址：

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

Linux 64位清华源地址：

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

假设需要安装的pytorch版本为：1.10.1
使用ctrl+f搜索关键字pytorch-1.10.1，结果如下：

pytorch-1.10.1-py3.6_cpu_0.tar.bz2
pytorch-1.10.1-py3.6_cuda10.2_cudnn7_0.tar.bz2
pytorch-1.10.1-py3.6_cuda11.1_cudnn8_0.tar.bz2
pytorch-1.10.1-py3.6_cuda11.3_cudnn8_0.tar.bz2
pytorch-1.10.1-py3.7_cpu_0.tar.bz2
pytorch-1.10.1-py3.7_cuda10.2_cudnn7_0.tar.bz2
pytorch-1.10.1-py3.7_cuda11.1_cudnn8_0.tar.bz2
pytorch-1.10.1-py3.7_cuda11.3_cudnn8_0.tar.bz2
pytorch-1.10.1-py3.8_cpu_0.tar.bz2
pytorch-1.10.1-py3.8_cuda10.2_cudnn7_0.tar.bz2
pytorch-1.10.1-py3.8_cuda11.1_cudnn8_0.tar.bz2
pytorch-1.10.1-py3.8_cuda11.3_cudnn8_0.tar.bz2
pytorch-1.10.1-py3.9_cpu_0.tar.bz2 1
pytorch-1.10.1-py3.9_cuda10.2_cudnn7_0.tar.bz2
pytorch-1.10.1-py3.9_cuda11.1_cudnn8_0.tar.bz2
pytorch-1.10.1-py3.9_cuda11.3_cudnn8_0.tar.bz2

根据pyx.x可知，可选的python版本有：3.6、3.7、3.8、3.9

2. 使用anaconda创建虚拟环境，并指定python版本

假设虚拟环境的名字为：pytorch101_gpu，python版本为：3.7

conda create -n pytorch101_gpu python=3.7

3. 在虚拟环境中安装cuda驱动

进入虚拟环境：

conda activate pytorch101_gpu 

换清华源：

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

首先需要手动安装cudatoolkit与cudnn
假设最终要安装的pytorch版本为pytorch-1.10.1-py3.7_cuda11.3_cudnn8_0.tar.bz2
可以确定cudatoolkit的版本为11.3，cudnn版本为8.0
假设安装的平台是Windows 64bit，则清华源地址为:

http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

安装命令如下：

conda install cudatoolkit=11.3 -c http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda install cudnn=8.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

注意：这里的cudatoolkit版本与机器实际安装的cuda版本无关

4. 安装pytorch

打开pytorch官网，查询pytorch版本与torchvision、torchaudio的对应关系：

https://pytorch.org/get-started/previous-versions/

查询到在Windows环境下，使用anaconda安装pytorch1.10.1的命令如下

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

可知pytorch=1.10.1对应torchvision=0.11.0，torchaudio=0.10.0
具体的安装命令如下：

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

参考

conda安装GPU版pytorch，结果却是cpu版本[找到问题根源，从容解决