torch.cuda.is_available()值为false,解决方案

torch.cuda.is_available() 值为true 就表示能用GPU进行加速训练
torch.cuda.is_available()值为false,解决方案_第1张图片如果是false,可能是驱动等级不够,怎么看驱动等级呢,如下
1.首先进入到这个目录下
C:\Program Files\NVIDIA Corporation\NVSMI
在这里插入图片描述

2.进入cmd
torch.cuda.is_available()值为false,解决方案_第2张图片

3.输入命令 nvidia-smi
torch.cuda.is_available()值为false,解决方案_第3张图片

通常版本大于390即可GPU进行加速
4.https://www.nvidia.cn/Download/index.aspx?lang=cn 下载对应的GPU驱动
torch.cuda.is_available()值为false,解决方案_第4张图片

下完后,在重复第一张截图步骤即可。

你可能感兴趣的:(机器学习,cuda,gpu)