torch.cuda.is_available() 值为true 就表示能用GPU进行加速训练
如果是false,可能是驱动等级不够,怎么看驱动等级呢,如下
1.首先进入到这个目录下
C:\Program Files\NVIDIA Corporation\NVSMI
2.进入cmd
3.输入命令 nvidia-smi
通常版本大于390即可GPU进行加速
4.https://www.nvidia.cn/Download/index.aspx?lang=cn 下载对应的GPU驱动
下完后,在重复第一张截图步骤即可。