pytorch的GPU为什么总是下载不成功

解决GPU无法使用

问题背景

不管是深度学习、神经网络，我们都经常使用pytorch，为了使模型训练加速，我们一般都会选择使用pytorch的CUDA版本，官方下载指令一般如下：

conda install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.1 -c pytorch -c nvidia

下载完成后验证：

import torch
print("PyTorch version:", torch.__version__)
print("CUDA version:", torch.version.cuda)
print("CUDNN enabled:", torch.backends.cudnn.enabled)
print("CUDA available:", torch.cuda.is_available())
if torch.cuda.is_available():
    print("Device:", torch.cuda.get_device_name(0))

明明下载的是GPU版本，但是却总是返回False。

问题解决

1. 删除原有环境

conda env remove --name yourenv

2. 新建环境

conda create -n yourenv python=3.11 -y
conda activate yourenv

修改下载指令

重点是== -c pytorch -c nvidia==，不要混入 conda-forge

 -c pytorch -c nvidia

类似下面

conda install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.4 -c pytorch -c nvidia

然后再次验证即可