- 최근 파이토치 모델 활용 중, 아래와 같은 오류가 지속적으로 발생함.
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'
'Deep Learning' 카테고리의 다른 글
PGGAN, StyleGAN 설명, Torch 코드 구현 (0) | 2024.02.19 |
---|---|
Pytorch Pix2Pix 모델 구현 (0) | 2024.02.16 |
conda env 실행 시 오류 해결 (0) | 2024.01.31 |
pytorch gpu(cuda) 정보 확인 (0) | 2024.01.17 |
anaconda 다중 사용자(multi-user) 환경 세팅 (0) | 2024.01.15 |