BUG：RuntimeError: CUDA error: invalid device ordinal CUDA kernel errors might be asynchronously repo
短信预约 -IT技能 免费直播动态提醒
报错分析
当运行以下代码报错:
# 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
免责声明：
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
