반응형
장고에서 cuda out of memory 가 날 경우 가 있다.
일단 원인을 찾아야 한다.
어떤 문제 때문에 cuda out of memory가 됬는지를 먼저 파악해야 하기 때문에 로그를 먼저 확인해야 한다.
로그를 확인하면서 어디에서 메모리를 많이 사용하는지 확인이 필요하다.
그중의 한가지 원인 은 :
tensorflow 모델을 사용할 경우:
tensorflow 모델을 사용할 경우에는 gpu메모리를 먼저 할당하여 사용하기 때문에
tensorflow 모델을 사용할 경우에는 cpu 모드로 한다음 pytorch 모델을 사용할 경우에는 gpu 모델로 바꾸면 된다.
반응형
'Deep learning > 오류해결' 카테고리의 다른 글
| 20201125- No module named 'png' (0) | 2020.11.25 |
|---|---|
| 20201125- IProgress not found. Please update jupyter and ipywidgets (0) | 2020.11.25 |
| 20201110-jsp image가 깨져서 나오지 않을 경우 (0) | 2020.11.10 |
| 20201110-Required request part 'file' is not present (0) | 2020.11.10 |
| spring-Exiting from "FORWARD" dispatch, status 404 (0) | 2020.11.09 |