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
测试:
完美!