今天在跑代码时出现了error:
RuntimeError: CUDA error: initialization error
cuda初始化出问题了,这是因为在python多线程跑gpu代码程序时先对cuda进行操作,然后在跑gpu代码时就没有cuda可用了。
在main的主程序代码加一行代码就可以了,用来获取cuda,在代码中只能使用一次:
import multiprocessing as mp
mp.set_start_method('spawn')
今天在跑代码时出现了error:
RuntimeError: CUDA error: initialization error
cuda初始化出问题了,这是因为在python多线程跑gpu代码程序时先对cuda进行操作,然后在跑gpu代码时就没有cuda可用了。
在main的主程序代码加一行代码就可以了,用来获取cuda,在代码中只能使用一次:
import multiprocessing as mp
mp.set_start_method('spawn')