解决torch.cuda.is_available()返回False的问题

解决torch.cuda.is_available()返回False的问题
解决torch.cuda.is_available()返回False的问题_第1张图片
(1)可能是电脑不支持CUDA
我的电脑-右键-设备管理器-显示适配器。如果没有NVIDIAN开头的则电脑没有独立显卡,不能支持CUDA。
解决torch.cuda.is_available()返回False的问题_第2张图片

(2)若有独立显卡,则检查GPU是否支持CUDA
在https://www.nvidia.cn/geforce/technologies/cuda/supported-gpus/查找自己电脑显卡型号是否支持CUDA
(3)检查显卡驱动版本
在anaconda prompt中进入pytorch环境,输入nvidia-smi,回车会显示驱动显卡版本。
解决torch.cuda.is_available()返回False的问题_第3张图片
CUDA9.2支持的版本是396.26以上,根据自己型号更新显卡驱动。进入https://www.nvidia.com/Download/index.aspx?lang=cn下载安装显卡驱动。

(4)有可能是pytorch和cuda版本不匹配。pytorch和cuda要有对应的关系,具体参考https://blog.csdn.net/weixin_43839245/article/details/108274605

你可能感兴趣的:(cuda,nvidia,pytorch)