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

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

第一步,安装cudn和cudnn,内容较长请看我的另外一篇博客文章。链接为:https://blog.csdn.net/my_name_is_learn/article/details/103913630

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

第二步,下载pytorch。如果在线安装的话慢得怀疑人生,所以推荐先下载好安装包再进行安装。安装包下载链接如下:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
打开之后挑选适合自己的安装包
在这里插入图片描述
第三步,安装pytorch,安装语句:
conda install --offline -n deep_image_matting2 pytorch-1.1.0-py3.6_cuda90_cudnn7_1.tar.bz2

语句解释:–offline表示离线安装。-n后面接的是虚拟环境的名字。最后面是安装包的目录文件名
在这里插入图片描述
第四步,安装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 torchvision==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

第六步,验证一下是否正确。看截图
在这里插入图片描述
成功结束完毕,点个赞呗看在你那么帅份上

发布了45 篇原创文章 · 获赞 24 · 访问量 3413

猜你喜欢

转载自blog.csdn.net/my_name_is_learn/article/details/104518141