马春杰杰博客
致力于深度学习经验分享!

RuntimeError: CUDA error: initialization error

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

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

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

如果你对这篇文章有什么疑问或建议,欢迎下面留言提出,我看到会立刻回复!

打赏
未经允许不得转载:马春杰杰 » RuntimeError: CUDA error: initialization error
蝴蝶PT招人啦

留个评论吧~ 抢沙发 评论前登陆可免验证码!

私密评论
  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址(选填,便于回访^_^)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏

切换注册

登录

忘记密码 ?

切换登录

注册