站点图标 马春杰杰

RuntimeError: CUDA error: initialization error

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

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

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

退出移动版