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