한동안 텐서플로우를 이용할 일이 없다가 다시 생겨 간만에 텐서플로우를 설치하게 되었다.
이미 검색해도 많이 나오는 내용이지만 기억할 겸 정리해두려고 한다.
Install Anaconda
우선 Anaconda를 설치하고 prompt 창을 실행한다.
Nvidia drivers
그리고 최신 버전의 Nvidia 드라이버를 설치한다. 설치가 안되어 있는 사람은 여기에서 다운로드 할 수 있다.
Virtual env
보통 가상환경(virtual env)을 이용한다. 가상환경에 대한 더 자세한 내용은 여기에서 확인할 수 있다.
Tested build configurations
나는 tensorflow 2.1.0 과 cudatoolkit 10.1을 사용했다. 이 외 버전들도 확인하고 싶다면 여기에서 확인 가능하다.
conda create -n tf-gpu tensorflow-gpu=2.1.0 cudatoolkit=10.1가상환경을 활성화하고 사용가능한 GPU를 확인해본다.
conda activate tf-gpu>>> import tensorflow as tf
>>> tf.config.experimental.list_physical_devices('GPU')