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