CUDA 11.8环境下torch.cuda.is_available()始终返回False求助
解决torch.cuda.is_available()返回False的方案
核心问题定位
你安装的PyTorch是基于CUDA 11.7编译的版本,而系统CUDA版本为11.8——尽管CUDA支持向下兼容,但部分场景下仍会出现适配异常,这大概率是问题根源。另外TensorFlow能正常调用GPU,说明系统层面的CUDA、CUDNN配置无问题,故障集中在PyTorch的版本适配环节。
具体解决步骤
1. 卸载现有PyTorch包
执行命令彻底移除当前PyTorch相关组件:
pip uninstall -y torch torchvision torchaudio
2. 安装适配CUDA 11.8的PyTorch版本
使用对应CUDA 11.8的安装命令:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
3. 验证安装有效性
安装完成后,在Python环境中运行以下代码确认:
import torch print(torch.__version__) print(torch.cuda.is_available()) print(torch.cuda.device_count())
若输出的torch.__version__包含cu118标识,且torch.cuda.is_available()返回True,则问题已解决。
4. 备选排查项(若上述步骤无效)
- 确认环境变量
CUDA_PATH指向CUDA 11.8的安装目录,而非其他版本 - 检查
PATH环境变量是否包含CUDA_PATH\bin和CUDNN的bin目录(CUDNN需放置在CUDA对应目录下或单独配置路径) - 重启终端或Python运行环境,确保环境变量修改生效
内容的提问来源于stack exchange,提问作者trunksabhi -
相关产品推荐
相关产品推荐

