conda install pytorch | GPU版本 | Linux系统 Windows系统

---

前言

尝试了网上各种安装pytorch办法的办法，但始终没有办法使用conda install的命令安装适用于GPU加速的pytorch版本。总结一些经验，在无可奈何之下使用手动安装最后竟然成功了。
特写此篇，给同样苦于安装pytorch的初学者提供一个方法。

---

提示：在使用命令行安装无果之后，才建议使用本教程方法。手动安装易出错。

一、conda install pytorch遇到的问题与已经尝试过的方法

1. 遇到的问题

比如：一直在collectingpackage metadata或solving environment

比如：报错 CondaHTTPError: HTTP 000 CONNECTION FAILED for url

本质上都是因为国内很难访问pytorch页面不成功或速度过慢，以致无法下载torch库。

2. 已经尝试过的办法

比如：更换清华镜像源

# 添加清华源的pytorch
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

比如：使用pytorch官方的对应命令，但删去 -c pytorch
pytorch官网
但这些方法都不行，进入手动安装环节！o(╥﹏╥)o

二、手动安装方案

进入到pytorch库各版本存放地址，主要下载torch和torchversion两个包。
网页地址如下：
torch包下载链接
torchversion包下载链接

1. 安装版本选择与下载

（1）首先，需要查看自身环境的python版本和cuda版本。

# 在conda环境中输入以下命令 查看python版本
python --version

# 在conda环境中输入以下命令 查看显卡版本
nvcc -V

如下图，我的python版本是3.9，cuda是11.6。

（2）根据自身python版本、cuda版本、系统选择对应安装包下载。
注意：有‘CPU‘字样的是只能在CPU框架下使用的，GPU框架必须选‘cu‘

此处需要特别注意！！！这点至关重要！！！
torchvision版本要比torch版本高1，例如：torch版本是1.12.0，那么torchvision版本要选0.13.0.
这一点依据是在多次使用pytorch官方命令时发现：

可以下载到本地或在linux系统下使用wget命令下载。

2. 安装命令

本文均是使用conda install命令安装，此处除外。这是因为.whl文件是python文件的压缩包，用pip install命令才能成功解压。

Tips： 需要cd进入到存放安装包的文件夹，才能解压下载。

先安装torch：

 # 实际安装时，需将.whl的名字更改为您所下载对应版本的文件名
 pip install torch-1.12.0+cu116-cp310-cp310-linux_x86_64.whl

后安装torchvision：

 # 实际安装时，需将.whl的名字更改为您所下载对应版本的文件名
 pip install torchvision-0.13.0+cu116-cp310-cp310-linux_x86_64.whl

三、检验安装成果

在conda输入python进入python，检验代码如下：

import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用 输出True则成功
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号

正确运行以上命令，代表你已经成功安装啦！
我的运行成功图如下：

---

总结

希望本文对大家有所帮助！如有不当之处，请您指出！如果对您有帮助，可以点个赞鼓励一下创作！(^▽)