马春杰杰 Exit Reader Mode

RuntimeError: CUDA error: initialization error

今天更换了一下卷积就这样了,还不清楚具体的原因,推测是pytorch多进程的问题。不过解决方案已经有了。

直接在主程序的开始加入下面这两句话即可:

import multiprocessing as mp
mp.set_start_method('spawn')

本文最后更新于2022年5月30日,已超过 1 年没有更新,如果文章内容或图片资源失效,请留言反馈,我们会及时处理,谢谢!