彻底解决torch、torchvision、torchaudio安装版本问题

彻底解决torch、torchvision、torchaudio安装版本问题

找版本

例如我安装的cuda版本是12.4
此次安装torch==2.1.0
在这里查询: torch官方版本
在此页面搜索：

有两个版本cuda，应该都能安装，我们安装cuda 12.1版本的，记录下版本
pytorch、2.1.0
torchvision、0.16.0
torchaudio、2.1.0

下载

在这里下载

找到cu121版本,cp38、cp39是python3.8、3.9，选择对应的版本
例如下载torch-2.1.0+cu121-cp38-cp38-win_amd64.whl
其他两个文件同理，搜版本，下载好

安装

在虚拟环境中，直接安装，其他同理

pip install D:\environmentAndsoftware\whl\torch-2.1.0+cu121-cp38-cp38-win_amd64.whl

验证

import torch
import torchvision
import torchaudio

# 检查CUDA是否可用
cuda_available = torch.cuda.is_available()
print("CUDA is available:", cuda_available)

# 打印PyTorch版本
print("PyTorch version:", torch.__version__)

# 打印TorchVision版本
print("TorchVision version:", torchvision.__version__)

# 打印TorchAudio版本
print("TorchAudio version:", torchaudio.__version__)

最后

pytorch下载速度很慢，耐心等待，下次创建环境就可以直接用了（python版本要一样）
安装出错，可能是python版本不一样
祝好运！！！