목록CUDA error (1)

RoBoLoG

[Error, Pytorch] RuntimeError: CUDA error: out of memory

Pytorch로 학습하는데 아래와 같이 오류 발생 : CUDA error: out of memory RuntimeError: CUDA error: out of memory CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1. 해결 방법: DataLoader에서 batch_size를 줄인다 or 더 좋은 GPU 구매..ㅎ
Error Solution/Pytorch 2024. 2. 27. 14:43
이전 Prev 1 Next 다음