20201112-django pytorch cuda out of memory

장고에서 cuda out of memory 가 날 경우 가 있다. 

일단 원인을 찾아야 한다.

어떤 문제 때문에 cuda out of memory가 됬는지를 먼저 파악해야 하기 때문에 로그를 먼저 확인해야 한다.

 

로그를 확인하면서 어디에서 메모리를 많이 사용하는지 확인이 필요하다.

 

그중의 한가지 원인 은 :

tensorflow 모델을 사용할 경우:

tensorflow 모델을 사용할 경우에는 gpu메모리를 먼저 할당하여 사용하기 때문에 

tensorflow 모델을 사용할 경우에는 cpu 모드로 한다음 pytorch 모델을 사용할 경우에는 gpu 모델로 바꾸면 된다.