Pytorch安装

Pytorch安装

Win10，anaconda3/python.exe（3.7版），使用的是anaconda里的python，不需要再自己安装python。
有GPU的pytorch的安装（2020-3-9）：
Pytorch的版本也一定要和CUDA匹配。Pytorch的所有历史版本：https://pytorch.org/get-started/previous-versions/，找到和CUDA对应的Pytorch和torchvision版本，不要直接就点最新的版本，看清楚再说。
我的版本是CUDA（10.0）+Pytorch（1.2.0）+torchvision（0.4.1）+pillow(6.0.0)
CUDA的安装：https://developer.nvidia.com/cuda-toolkit-archive：这个是英伟达CUDA的历史版本，找到需要的版本号，不要直接下载最新版的安装包。下载安装包后一切默认安装即可。安装完成后可通过cmd里面输入nvcc -V查看CUDA版本号。
cuDNN的安装：英伟达官网下载: https://developer.nvidia.com/rdp/cudnn-download 下载对应的版本，需要注册，可以用微信号或者qq注册。
下载完成解压缩之后，把bin，include，lib/x64里的文件分别放入CUDA对应的bin，include，lib/x64里面。CUDA的默认路径为：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
Pytorch的安装：官网过于慢，而且容易断，清华源我真的也装不上，不清楚为啥，所以下载pytorch的whl文件，用pip安装，打开官网所有pytorch的所有whl文件链接：https://download.pytorch.org/whl/cu100/torch_stable.html下载相应文件，torch后面是pytorch版本号，cp后面是python版本号。

下载速度相当慢，下载完成之后，pip安装whl文件：

pip install torch-1.2.0-cp37-cp37m-win_amd64.whl

提示还需要下载numpy，ctrl+c终止，自行安装numpy，豆瓣源安装numpy：

pip install numpy==1.16.1 -i https://pypi.douban.com/simple

重新再次pip安装whl文件，成功后，豆瓣源安装torchvision:

pip install torchvision==0.4.1 -i https://pypi.douban.com/simple

测试发现找不到PILLOW_VERSION，但是之前其实已经pip install pillow，发现原因是pillow7.0.0 后的版本就没有 PILLOW_VERSION ，所以需要重新安装＜7.0.0版本：

pip install pillow==6.0.0 -i https://pypi.douban.com/simple

测试：

完美！