【百分百胜率PyTorch安装教程】安装指定pytorch, python, cuda版本的gpu版pytorch

两步完成安装：

1. 搜索需要的包
2. 安装需要的包

本文以安装pytorch版本2.1.2，python版本3.10，cuda版本11.8的GPU版pytorch为例

1. 使用conda search在pytorch官方channel中找到对应版本的pytorch

• conda search pytorch : 搜索pytorch这个包
• -c pytorch : 在pytorch这一channel下搜索
• | grep : linux系统中搜索指定的字符串模式，这里用来过滤搜索结果，找到指定pytorch, python, cuda版本的gpu版pytorch

conda search pytorch -c pytorch | grep cuda11.8 | grep 2.1.2 | grep py3.10

输出（搜索）结果如下:

pytorch                        2.1.2 py3.10_cuda11.8_cudnn8.7.0_0  pytorch

2. conda install 指定安装版本和安装包

# 这一步是为了确保安装时使用pytorch channel而不使用其他channel,可以先跳过这一步试试
conda config --remove-key channels

# 在安装pytorch时指定安装上面搜索结果中的版本号2.1.2和包py3.10_cuda11.8_cudnn8.7.0_0
# 同时安装torchvision torchaudio，会自动安装匹配版本
# pytorch-cuda=11.8 -c nvidia用于安装指定cuda版本
conda install pytorch=2.1.2=py3.10_cuda11.8_cudnn8.7.0_0 torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia -y

成功！

cuda安装参考