问题就在于训练所使用的num_workers数量问题,直接将
num_workers=0
就不会报错了
RuntimeError: Cannot re-initialize CUDA in forked subprocess
最新推荐文章于 2022-10-17 16:24:07 发布
问题就在于训练所使用的num_workers数量问题,直接将
num_workers=0
就不会报错了