Python 官方文档:入门教程 => 点击学习
报错分析 当运行以下代码报错: # self.opt.gpu_ids = ["1"]torch.cuda.set_device(self.opt.gpu_ids[0]) 报错信息如下: Runtime
当运行以下代码报错:
# self.opt.gpu_ids = ["1"]torch.cuda.set_device(self.opt.gpu_ids[0])
报错信息如下:
RuntimeError: CUDA error: invalid device ordinal
CUDA kernel errors might be asynchronously reported at some other api call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
报错完整截图:
报错的信息告诉我们,编号“1”是无效的设备序号。但我使用的设备属于单机多卡,是有编号为"1"的显卡的。
检查报错代码前面执行过的程序,特别是导入第三方库部分,发现利用os
库指定了该程序可见的GPU编号及数量,即:
import osos.environ["CUDA_VISIBLE_DEVICES"] = "1"
因此,注释掉os.environ["CUDA_VISIBLE_DEVICES"] = "1"
后,重新运行程序,顺利解决bug~
来源地址:https://blog.csdn.net/qq_41813454/article/details/129774989
--结束END--
本文标题: BUG:RuntimeError: CUDA error: invalid device ordinal CUDA kernel errors might be asynchronously repo
本文链接: https://www.lsjlt.com/news/391527.html(转载时请注明来源链接)
有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
2024-03-01
2024-03-01
2024-03-01
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
2024-02-29
回答
回答
回答
回答
回答
回答
回答
回答
回答
回答
0