ERROR
[DL] RuntimeError: Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing, you must use the 'spawn' start method
파송송
2023. 4. 25. 12:06
728x90
multi GPU를 사용하려고 할 때 생기는 에러이다.
torch.multiprocessing.set_start_method('spawn')
spawn을 지정해주면 해결된다.
728x90