PyTorch提示CUDA不可用问题排查与解决（NVIDIA T500+Win11）

环境信息 #

• 显卡：支持CUDA的NVIDIA T500
• 操作系统：Windows 11
• PyTorch安装命令：conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
• CUDA相关版本：
  • nvcc --version 输出CUDA版本11.6
  • nvidia-smi 显示驱动版本526.67、CUDA版本12.0

问题现象 #

在Jupyter Notebook中执行以下代码，返回结果为False：

import torch
torch.cuda.is_available()

排查结果 #

通过conda list查看已安装包，发现pytorch-mutex是cpu版本，与安装时指定的CUDA版本不匹配。

解决步骤 #

1. 卸载pytorch-mutex：

   conda uninstall pytorch-mutex
   

2. 等待Anaconda自动修复依赖不一致问题
3. 重新执行原PyTorch安装命令：

   conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
   

4. 验证：执行conda list确认pytorch-mutex变为cuda版本，此时torch.cuda.is_available()返回True，问题解决。

内容的提问来源于stack exchange，提问作者Matthias