首页 » 机器视觉 » 正文

日常Debug——No CUDA GPUs are available

2024-07-14 224 0

报错如下：
No CUDA GPUs are available
解决方法：
1、首先在报错的位置net.cuda前加入cuda检测语句：

print(torch.cuda.is_available())

输出为False，证明cuda不可用
2、检查本机中的cuda是否安装成功，且版本号是否与pytorch的版本号对应。检查发现没有问题
3、检查os.environ["CUDA_VISIBLE_DEVICES"] = "1"语句，将1改为0，再运行无误。原因可能是本机上只有一个cuda，序号应为0。

版权声明：本文为CSDN博主「Lemon-Ning」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_42103801/article/details/116675059