Windows平台上安装pytorch很容易出错,查了各种资料,都是写的不详细不完整,对于小白来说,越看越糊涂,经历了艰辛的安装历程,所以在这我想写个详细易懂的教程,分享一下我的经验。
1.首先需要安装Anaconda,下载地址https://www.anaconda.com/distribution/#windows,选择3.7版本,下载并安装即可。
这步一定要注意两个方框一定要都选,就是加入到系统环境里,如果第一个不选后面肯定会报错。其他的都是Next即可。
接下来,如果有英伟达显卡,进入第2步;没有显卡直接到第4步。
2. 接下来安装CUDAToolkit,最新的CUDA是10.0版本,我的显卡比较老了,我选择8.0。转到该网址“https://developer.nvidia.com/cuda-80-ga2-download-archive,选择对应的参数,下方会有对应的CUDA文件。例如,下图是我选择的版本,有1.3G大小,下载下来
如果你想找其他版本的CUDA,请前往:https://developer.nvidia.com/cuda-toolkit-archive
运行.exe开始安装,过程会稍微久一点,耐心等待就好了,中途会提示你是否安装NVIDA显示适配器,选择确认。
3.cudnn下载, 点击https://developer.nvidia.com/rdp/cudnn-archive进入下载,cudnn的下载需要注册,下载符合自己要求的版本,我选择的是cuDNN v7.14 Library for Windows 7,下载过程比较漫长,下载可能失败,需要多下载几次。
下载得到的是一个压缩包,解压缩以后得到的是cuda文件夹,里面包含bin、include、lib三个文件夹,将这三个文件夹里面的文件复制到CUDA的安装路径下同名文件夹内即可,复制完成后就相当于完成了安装。
4.pytorch安装
首先进入官网https://pytorch.org/,打开官网后选择对应的选项:
打开cmd,运行红圈里的命令,如果你和我一样,采用8.0CUDA,3.7版本Python,输入以下命令:
conda install pytorch torchvision cuda80 -c pytorch,然后回车,系统会自己安装
如果电脑没有英伟达显卡,那就只能安装CPU版本的,CUDA选择None,在cmd运行:
conda install pytorch-cpu torchvision-cpu -c pytorch
我第一次安装出现An HTTP error occurred when trying to retrieve this URL. 的错误 ,于是各种查资料,尝试各种办法,也尝试使用网上说的清华镜像源,但是会报错,可能我操作的不对,最后发现,再运行一遍刚才的命令就好了,即
C:\Users\Administrator>conda install pytorch torchvision cuda80 -c pytorch
我觉得不使用清华镜像源下载的也挺快的,建议在网好的地方就可以。
5.验证安装是否成功
验证pytorch是否安装成功,在终端输入:python,此时进入python环境。
然后,继续输入import torch
torch.__version__ ,然后回车不报错就表明安装成功。