您当前的位置:首页 > 计算机 > 软件应用 > 开发工具(IDE)

win10下使用anaconda安装pytorch和torchvision

时间:03-12来源:作者:点击数:

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

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的全部过程,供大家参考。

附:

现在pytorch已经到1.6了,我之前尝试安装过1.4,但是并不能使用gpu。查询之后发现,是因为cuda版本太低了。现在我把cuda版本,Torch版本,Torchvision版本以及python版本的对应关系做了一个表,仅供大家参考(可能存在一些不当)。

在这里插入图片描述
在这里插入图片描述

2020年12月9日

我一个同学买的3090,安装对应版本的驱动+CUDA11.1+对应CUDnn+PyTorch1.7,测试可用。

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门