conda虚拟环境中搭建pytorch+cuda+cudnn

conda虚拟环境中搭建pytorch+cuda+cudnn

以下各步均需在激活的虚拟环境中进行

1. conda创建并激活虚拟环境 例：conda创建py36

conda create -n py36 python=3.6
activate py36

2. 安装pytorch框架以及对应的cuda&cudnn，具体型号匹配可以到pytorch官网查询，换源清华源,

问题1:在conda中如果利用pytorch中的链接换用清华园直接下载pytorch+cuda会出现找不到pytorch安装包的情况，

解决1：pip install torch，不使用pytorch官网的链接（这种方法会自动安装最新版本的pytorch，最好指定版本以和cuda以及cudnn匹配）
解决2：首先从清华源安装对应版本cuda以及cudnn，接着使用pip安装pytorch

方法2对应命令行如下：

conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

对应网址如下：cuda,cudnn,pytorch.
可能出现的问题：在安装cudnn7.6.5时其默认会安装2.7版本的pip，因此在安装cudnn后最好再重新安装python，否则安装pytorch时会报错。命令行如下所示
conda install python=3.6

3.将创建的虚拟环境导入到jupyter中，代码如下
pip intall ipykernel
python -m ipykernel install --user --name py36 --display-name py36
-name py36 --display-name py36`
4.在jupyter中将kernel换成py36即可。

最后验证是否安装成功，代码如下

activate py36
python
import torch
torch.cuda.is_available()

最终输出：true说明安装成功