win10下使用anaconda安装pytorch和torchvision

一、系统配置以及软件版本

1、操作系统:win10
2、显卡:NVIDIA GeForce GTX 1070Ti
3、cuda_9.0.176_win10
链接:https://pan.baidu.com/s/1f9MowahErE9u60LO1MOcPw
提取码:5k2c
4、cudnn-9.0-windows10-x64-v7.1
链接:https://pan.baidu.com/s/1MGec2JIhAYV9GG0T-4MBnw
提取码:l0ua
5、Anaconda3-4.3.0-Windows-x86_64
链接:https://pan.baidu.com/s/13PA5ZdWWibsQ8acr6SZjow
提取码:kae8
PyTorch官网上会根据你提供的相关版本告知你如何安装相应的pytorch。
在这里插入图片描述

二、安装过程

1、如果需要安装GPU版本的,首先安装CUDA,顺序执行安装程序即可,随后安装cudnn,将cudnn解压,对应文件复制到cuda安装文件夹下即可。
2、安装anaconda4.3,没有什么难度,选择安装自带3.6版本的python。
3、打开命令提示符(cmd),输入:conda --version,出现anaconda版本信息。输入:conda info --envs,出现安装的环境(初始时只有root环境,我的已经安装了TensorFlow和pytorch)。
在这里插入图片描述
4、创建pytorch环境,输入:conda create -n pytorch python=3.6,“-n”是“name”的缩写,pytorch是你创建的环境的名称,其实可以自己命名,后面是在环境里安装python,前面anaconda里也装了python,那是在root环境(主环境)安装的。出现你将要安装的包名和版本信息,输入y,继续安装。
在这里插入图片描述
5、安装成功后,出现如下图所示信息,输入:activate pytorch,进入pytorch环境里面,这个时候pytorch环境相当于只是个空的框架,真正的pytorch还没有装上。(输入:deactivate pytorch,退出pytorch环境)
在这里插入图片描述
6、如上图所示,在pytorch环境中,输入:conda install pytorch -c pytorch,开始安装pytorch,同样需要输入y,继续安装。
7、安装完成后,输入:python,显示python版本信息,并进入到python里,输入:import torch,没有任何提示则说明安装成功。(我这里在没安装torchversion的情况下,以为安装了pytorch就可以,import torchvisio结果提示错误)
在这里插入图片描述
8、输入:exit(),退出python,进入到pytorch环境,输入:conda install torchvision -c pytorch,结果显示没有包。
在这里插入图片描述
9、那我们换一种安装方式,输入:pip install torchvision,会看到安装信息,最后出现“Successfully……”,说明安装成功。然后我们重复步骤7,就显示torch和torchvision都安装成功。
在这里插入图片描述
以上就是安装pytorch的全部过程,供大家参考。

发布了17 篇原创文章 · 获赞 20 · 访问量 2万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览