에러 내용
~~~~,
torch._C._cuda_init()
RuntimeError: cuda runtime error (3) : initialization error at ../aten/src/THC/THCGeneral.cpp:54
python3.8에서 multiprocessing을 활용해서 할 경우, 이런 에러가 생긴다.
이럴 경우,
if __name__ == '__main__' :
문 안에 multiprocessing.set_start_method('spawn') 코드를 넣어주면 된다.