Windows에서 TF 1.5 GPU버전 설치방법

Windows에서 TF 1.5 GPU버전 설치방법

2018/2/1자 Windows 10, TF 1.5, Anaconda3 Python3.6, CUDA 9.0, cuDNN v7.0.5에 Tensorflow GPU 설치방법

테스트환경 : AMD Ryzen 1800X, NVIDIA Titan X(p)

1. Tensorflow 1.5에서 CUDA 9.0을 요구. CUDA 9.1로 실행시 오류발생. 그래픽드라이버를 최신버전으로 따로 설치 후에 사용자설치로 CUDA 9.0만 설치권장.

2. Anaconda3으로 진행해도 무방. TF Windows 안내에 3.5 & 3.6지원이라 써있음. 3.5버전을 찾는다면 Anaconda3 4.2버전설치를 권장. https://repo.continuum.io/archive/index.html

3. Download cuDNN v7.0.5으로 진행함. 라이브러리라서 cuDNN v7.0.4와 버전관계는 없는듯.

4. 설치는 Tensorflow 1.5로 진행.(2018년 2월1일 현재 nvidia-docker 지원이 안되서 Windows 환경에서 Docker는 TF CPU버전밖에 지원이 안됨)

5.https://www.tensorflow.org/install/install_windows

위에 TF 설치 URL중 Anaconda에 설치하는 법 참조

(tensorflow)C:> pip install --ignore-installed --upgrade tensorflow-gpu

6. Python 실행후 TF 테스트코드 실행

ps. AMD Ryzen CPU의경우 TF 코드실행시 warning이 발생될 수 있음. 아래코드를 Session 실행 전에 넣어주면 warning이 사라짐.

# Just disables the warning, doesn't enable AVX/FMA

import os

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'