报错解决：RuntimeError: CUDA error: out of memory CUDA kernel errors

作者：从前慢现在也慢 | 2024-03-11 21:40:17

踩

报错现象

在device为CUDA的情况下，运行多个程序的时候，可能会遇到如下报错

RuntimeError: CUDA error: out of memory 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

分析原因

程序默认都在同一个卡上跑，所以会出现内存不足的情况

解决

在代码开头加上如下代码，为该代码的运行指定一个可用的卡

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
1
2

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/220662