windows10下pytorch1.4.0安装

安装步骤如下：
1.打开Anaconda Prompt，输入：

conda create -n pytorch1.4.0 python==3.6

创建一个pytorch的虚拟环境。

2.激活环境：

conda activate pytorch1.4.0

3.安装CUDA和Cudnn

conda install cudatoolkit==10.1

conda install cudnn==7.6.5

如果网不好的可以下载离线包：
cudatoolkit10.1:
链接：https://pan.baidu.com/s/1V9QVXn0E9uruWKwk9CDRuQ
提取码：1234

cudnn7.6.5:
链接：https://pan.baidu.com/s/1rGshlMTZgKNzbccMvXET6g
提取码：1234

下载之后放在创建的虚拟环境之下：

然后安装：

conda install cudatoolkit-10.1.168-0.tar.bz2
conda install cudnn-7.6.5-cuda10.1_0.tar.bz2

4.下载pytorch本地包

链接：https://pan.baidu.com/s/1ayKaBMlzFijudYtTxWD28A
提取码：1234

在安装之前，先安装依赖包：

pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install six -i https://pypi.tuna.tsinghua.edu.cn/simple

把刚才下载的包放在虚拟环境之下：

然后安装：

pip install torch-1.4.0-cp36-cp36m-win_amd64.whl
pip install torchvision-0.5.0-cp36-cp36m-win_amd64.whl --no-deps

注意安装torchvision的时候要加上 –no-deps
然后等待安装完毕即可。

5.测试是否安装成功
输入python进入Python的环境，
然后输入import torch，如果没报错即安装成功。
如果是python3.6.0可能会遇到这个错误：

这时候可以输入quit()退出Python环境，然后升级一下python版本：

conda install python==3.6.5

等待安装完毕。

然后再进入python环境，同样import torch，没报错则继续：

print(torch.cuda.is_available())

最后返回true即说明GPU可用，安装成功。