TensorFlow2 Tensorflow GPU 메모리 할당 제어 Pycharm에서 Tensorflow를 돌릴 때 기본적으로 1개의 Python만 실행이 된다. 여러 개 Python 파일을 실행하고 싶으면 CPU / GPU / GPU 메모리 등을 확인해야 한다. 우선 기본적으로 1개의 python 파일을 tensorflow를 사용할 때는 다음과 같다. * watch nvidia-smi 확인하기 watch nvidia-smi GPU는 56%, GPU 메모리는 7699MB가 할당된 것을 확인할 수 있다. 메모리를 7699MB가 할당된 이유는 디폴트로 전체가 할당되기 때문이다. 그래서 2개 이상의 tensorflow를 돌리기 위해서는 메모리가 부족하다. Python Code에서 GPU 메모리를 지정 혹은 최소로 설정을 해서 실행해보자. # 메모리 최소 할당 config = .. 2020. 9. 4. Ubuntu Tensorflow nvidia driver 설치 1. 설치 sudo add-apt-repository ppa:graphics-drivers/ppa sudo apt update sudo apt-get install nvidia-driver-440 2. 재부팅 sudo reboot 3. 확인 nvidia-smi CUDA 설치 1. CUDA Homepage에서 CUDA Toolkit 10.0 (Sept 2018) 선택 URL : https://developer.nvidia.com/cuda-toolkit-archive CUDA Toolkit Archive Previous releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can .. 2020. 4. 7. 이전 1 다음
