CUDA out of memeory.. 장애 발생 해결 방법
Devel/Deep Learning2024. 5. 3. 15:40CUDA out of memeory.. 장애 발생 해결 방법

[장애 메세지]If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF [해결 방법]다음과 유사한 방식으로 환경 변수 설정Windows: set 'PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512'Linux: export 'PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512'

image