Conda环境安装pytorch全流程

一、pytorch安装一般问题解析

        首先安装pytorch最大的问题就是pytorch,cuda和自己电脑显卡这三个版本兼容的问题，导致存在安装了pytorch却无法使用显卡跑代码的问题。下面我将逐步指导。

二、安装步骤

1、检查自己电脑英伟达显卡最高支持的cuda版本是多少

在命令行输入nvidia-smi，检查自己显卡最高支持的cuda版本是多少，原则上你下载cuda的版本只要不高于这个值就可以。

2、查看下载源是否有合适的pytorch安装包

注：由于网络原因，绝大部分使用pytorch官网下载pytorch安装包会失败，所以我们使用清华镜像源进行下载，pytorch清华源地址：Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror我们先提前在清华源上找到合适安装包，也就是提前把python,pytorch,cuda这三者的版本确定好。

比如我标注的这个安装包显示，pytorch的版本是1.11.0，python的版本是3.8，cuda的版本是11.3（低于我电脑最高可支持的版本）。这几个版本结合自己的电脑配置以及需求进行匹配即可。

3、安装cuda

首先进入cuda官网然后点击下载

然后点击以前版本

然后找到自己刚才决定的cuda版本

然后根据自己的系统选择下载，最后无脑安装cuda即可。

紧接着在命令行测试一下cuda就安装成功了：nvcc -V

4、在环境中安装pytorch

在conda中新建一个虚拟环境，或者在已经存在的环境中安装pytorch也行，但需要注意的是，这个环境中的python版本需要是你最开始选择好的版本，我这里最开始选择的是python=3.8。

创建好环境之后，现在去pytorch官网找到我们之前选择好的pytorch版本。这里选择去查看以前的版本。

然后找到pytorch的版本我这里是11.1，以及对应匹配的cuda版本这里是11.3。然后复制下面的安装代码

 然后再环境中使用这个代码安装pytorch，这里需要修改一下下载源也就是-c后面的部分，修改为我们之前查看pytorch安装包的清华源镜像地址。

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

然后执行就好，一般只要网络不是特别恼火，pytorch就会成功安装，最后测试pytorch是否可用。

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())  #输出为True，则安装无误。

 

这里输出True就万事大吉了。