Windows11下Anaconda安装PyTorch+CUDA失败,torch.cuda.is_available()为False
PyTorch CUDA不可用问题:驱动版本与CUDA兼容性解析
你的理解存在偏差,不需要将GPU驱动降级至支持CUDA 12.1的版本。NVIDIA驱动对CUDA Toolkit是向下兼容的——只要驱动版本满足对应CUDA版本的最低要求即可。CUDA 12.1要求的最低驱动版本是530.30.02,你的551.23版本远高于该阈值,完全可以支持CUDA 12.1,因此驱动版本不是问题根源。
以下是针对你当前情况的排查步骤:
确认PyTorch安装的是CUDA版本而非CPU版本
- 激活你的Anaconda环境,执行
conda list或pip list,查看torch版本是否带有CUDA标识(例如2.2.0+cu121,而非cpu后缀的版本)。 - 若之前安装过CPU版本,需先彻底卸载:
conda uninstall pytorch torchvision torchaudio -y # 或用pip卸载 pip uninstall torch torchvision torchaudio -y - 重新执行PyTorch官方提供的对应CUDA版本安装命令(以CUDA 12.1为例):
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
- 激活你的Anaconda环境,执行
检查环境变量与环境激活状态
- 确保已卸载系统级CUDA后,删除环境变量中残留的
CUDA_PATH、CUDA_HOME等相关条目。 - 确认当前激活的Anaconda环境是你安装PyTorch的环境,避免误调用其他环境中的CPU版本PyTorch。
- 确保已卸载系统级CUDA后,删除环境变量中残留的
基础可用性验证
执行以下命令,输出关键信息用于排查:import torch print(f"PyTorch内置CUDA版本: {torch.version.cuda}") print(f"CUDA是否可用: {torch.cuda.is_available()}") print(f"GPU设备数量: {torch.cuda.device_count()}")若
torch.version.cuda显示的版本与你安装时选择的CUDA版本不符,说明安装过程存在问题。其他排查方向
- 尝试覆盖安装当前的551.23驱动,修复可能存在的驱动损坏问题。
- 检查Windows Defender或第三方杀毒软件是否拦截了PyTorch的CUDA相关组件。
内容的提问来源于stack exchange,提问作者David Airapetyan
相关产品推荐
相关产品推荐

