天天看点

一分钟搞懂在Windows10系统安装使用pytorch框架

该文章将记录如何在Windows10系统安装制定版本的pytorch框架,以便使用上电脑的独立显卡GPU。

第一步,安装cudn和cudnn,内容较长请看我的另外一篇博客文章。

注意,从“第三件事步骤”开始安装就行。我自己安装的版本是cudn9.0和cudnn7.1版本。该篇文章也会以这个版本为例。

第二步,下载pytorch。如果在线安装的话慢得怀疑人生,所以推荐先下载好安装包再进行安装。安装包下载链接如下:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

打开之后挑选适合自己的安装包

一分钟搞懂在Windows10系统安装使用pytorch框架

第三步,安装pytorch,安装语句:

conda install --offline -n deep_image_matting2 pytorch-1.1.0-py3.6_cuda90_cudnn7_1.tar.bz2语句解释:–offline表示离线安装。-n后面接的是虚拟环境的名字。最后面是安装包的目录文件名

一分钟搞懂在Windows10系统安装使用pytorch框架

第四步,安装pytorch的依赖包torchvision。pytorch为1.1.0版本的时候,torchvision对应的版本为0.3.0.具体的版本匹配信息可以看下面链接的说明:https://pytorch.org/get-started/previous-versions/

安装语句为:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torchvisinotallow==0.3.0

第五步,其实完成到这一步就完成所有事情了,可是个人电脑出现Pillow版本不对的现象(找不到Pillow Version),所以我就卸载了Pillow7.0.0版本的Pillow而重新安装了Pillow5.3.0版本的Pillow.附上卸载和安装的语句:

pip uninstall Pillow

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple Pillow==5.3.0

第六步,验证一下是否正确。看截图

一分钟搞懂在Windows10系统安装使用pytorch框架

继续阅读