RuntimeError: CUDA error: initialization error 如何

今天在跑代码时出现了error:

RuntimeError: CUDA error: initialization error

cuda初始化出问题了,这是因为在python多线程跑gpu代码程序时先对cuda进行操作,然后在跑gpu代码时就没有cuda可用了。
在main的主程序代码加一行代码就可以了,用来获取cuda,在代码中只能使用一次:

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

你可能感兴趣的:(深度学习,pytorch)