大家好，又见面了，我是你们的朋友全栈君。
一、系统配置以及软件版本
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
发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/152579.html原文链接：https://javaforall.net
