关于RuntimeError: Cannot re-initialize CUDA in forked subprocess和CUDA error: initialization error的解决

狂小虎

已于 2022-09-29 15:30:49 修改

阅读量2.9k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Deep Learning 文章标签：深度学习 pytorch python

于 2022-09-27 17:06:00 首次发布

本文为博主原创文章，未经博主“狂小虎”本人允许不得转载。

本文链接：https://blog.csdn.net/Davidietop/article/details/127072346

在尝试在GPU上运行深度学习任务时，遇到了CUDA error: initialization error和RuntimeError: Cannot re-initialize CUDA in forked subprocess的问题。解决方法是在DataLoader之外将数据转移到CUDA上，避免在Dataloader内部操作CUDA。通过修改代码，将数据加载和CUDA操作分离，成功解决了这两个错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近小虎在cpu上试跑了detection任务的程序没什么问题，但是在gpu上运行时发生了关于cuda的一些报错，发现解决问题的精华就一句话："不可在DataLoader或DataSet内将任何数据放到CUDA上，而是等到程序运行出DataLoader之后（也就是到了train里的时候）将数据放到CUDA上。"下面给出问题解决的例子和代码。

问题CUDA error: initialization error

先看看小虎程序的报错，报错中说明了在对bounding box的左上角加长宽坐标形式转成左上右下对角点坐标形式时，CUDA初始化回发生错误。

  File "/home/wei/lef/lef2/lib/model/detection/matcher/utils.py", line 173, in centreForm2CornerForm
    boxes[..., :2] -

最低0.47元/天解锁文章

200万优质内容无限畅学