텐서플로우 설치 - windows
-
Python 3 설치 - anaconda 사용
https://www.anaconda.com/distribution/ 에서 anaconda installer 다운로드 및 설치 -
NVIDIA GPU driver 설치
https://www.nvidia.co.kr/Download/index.aspx?lang=kr -
NVIDIA CUDA Toolkit 설치
https://developer.nvidia.com/cuda-downloads -
cuDDN 설치
https://developer.nvidia.com/cudnn
다운받은 압축파일을 적당한 위치에 푼다. 여기서는 C:\tools/cuda 에 풀겠다. -
환경변수 추가
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\libx64;
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include;
C:\tools\cuda\bin;
해당 경로들을 환경변수에 추가한다. - 텐서플로우 설치 - pip
$ pip install tensorflow // 텐서플로우2 버전부터는 해당 명령어로 CPU와 GPU를 모두 설치해준다
- 설치 확인
$ python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))" > 2020-04-21 20:36:09.675162: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2 tf.Tensor(-342.69507, shape=(), dtype=float32)
결과가 나오면 설치가 잘 된것이다. 위에 메시지는 내 cpu는 더 많은 명령어를 지원하지만, pip로 받은 텐서플로우 빌드 버전에서는 해당 명령어를 사용하지 못한다는 의미이다. 만약 해당 메시지가 싫거나 좀 더 빠른 동작을 원한다면 텐서플로우 소스를 받아서 빌드하면 된다.