728x90

메모리가 모자라서 생기는 문제였다.

필요없는 메모리를 삭제하기로 결정했다.

sudo fuser -v /dev/nvidia*

cuda를 쓰고 있는 메모리들 확인이 가능하다.

sudo kill -9 PID

안쓰는 메모리들을 kill해주면 쓸 수 있는 메모리가 많아져 다시 코드를 돌릴 수 있다.

728x90

'Pytorch' 카테고리의 다른 글

VS Code에서 pytorch 사용을 위한 python 환경 설정  (0) 2022.12.16
728x90

보통  pytorch나 tensorflow 기반의 딥러닝 학습을 할 때 anaconda를 많이 쓰는 편이다.

이미 pytorch나 anaconda 등이 서버에 깔려 있음을 가정하고 작성한 글이다.

 

vs code의 경우에는 나이브한 python 그 자체로 interpreter경로가 세팅되어 있는 경우가 많아서

torch의 import가 안 먹는 경우가 있다.

 

아니 pytorch는 분명 설치했는데 왜 그러지 할 때에는 python의 경로 설정 잘못인 경우가 많다.

 

이런 경우, Python interpreter 설정을 변경해주어야 한다.  

 

F1을 누르고 <Python: Select Interpreter>를 선택하여

 

Python으로 되어 있는 interpreter 경로를 pytorch가 설치되어 있는 interpreter로 경로를 바꿔주면 된다.

(내 경우에는 두 번째에 있는 경로를 선택)

728x90

'Pytorch' 카테고리의 다른 글

Linux command로 cuda out of memory 해결  (0) 2023.06.12

+ Recent posts