[TORCH_UECUDA_DSA] CUDA 오류

• 최근 파이토치 모델 활용 중, 아래와 같은 오류가 지속적으로 발생함.

RuntimeError: CUDA error: device-side assert triggered
Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.

 

• os 모듈을 활용해서 아래와 같은 명령어를 python script에 추가하여 문제 해결

os.environ['CUDA_VISIBLE_DEVICES'] = '1'
os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
os.environ["TORCH_USE_CUDA_DSA"] = '1'