RoBoLoG

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

Error Solution/Pytorch

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

SKJun 2024. 2. 27. 14:43

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 구매..ㅎ

728x90
반응형

'Error Solution > Pytorch' 카테고리의 다른 글

[Pytorch] RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking arugment for argument mat2 in method wrapper_mm)  (0) 2024.03.04
[Pytorch] AttributeError: module 'torch' has no attribute 'frombuffer'  (1) 2024.02.27
'Error Solution/Pytorch' Related Articles more