1、报错:No CUDA GPUs are available
2、参考了以下两位博主的文章,已解决,深深感谢!附上链接云天徽上的文章、春风知江南
3-1、检查CUDA是否正确安装
打开cmd(win+R,输入cmd打开),激活环境 ,验证代码。
avtivate 环境名
nvcc --version
可以看出,cuda已经正确安装,并且版本为11.8
3-2、检查cuda和torch版本是否匹配
我的cuda版本是11.8,torch版本是是2.0.0,理论上匹配没问题,谨慎起见,用代码验证一下
结果为True,说明版本匹配没问题。
3-1、查看GPU设备数量
命令返回1,说明是有GPU设备的。
3-1、报错解决
再次运行代码时,报错已经解决了,根据博主春风知江南的说法,cuda使用之前都要检测一下cuda是否可用,没检测就默认没有,至此,解决完此bug。