본문 바로가기
인터넷/모바일/머신러닝

윈도우10에서 텐서플로우 설치하기.

by blade. 2017. 8. 8.


윈도우10 64비트 버전에서만 실행되므로, 32비트를 가지고 있다면, OS 옆그레이드 먼저...

필요한 프로그램 : python, anaconda, cuda, cuDNN

모든 프로그램들의 버전이 중요하므로, 아래 가이드에 나오는 해당 버전을 받아서 설치해야함.

각각 프로그램의 다운로드 및 설치방법.


1) Python 3.6.1 설치

https://www.python.org/downloads/release/python-361/


2) Anaconda3-4.4.0 설치

Anaconda3-4.4.0-Windows-x86_64.exe

덩치가 큰 프로그램이기 때문에 약간의 인내심을 요함.

https://docs.continuum.io/anaconda/install/hashes/Anaconda3-4.4.0-Windows-x86_64.exe-hash


3) CUDA 8.0 설치

cuda_8.0.61_win10.exe

만약 GPU가 없다면, (3)번과 (4)번 인스톨은 패스.

https://developer.nvidia.com/cuda-downloads


4) cuDNN 5.1 for CUDA 8.0 설치

cudnn-8.0-windows10-x64-v5.1.zip

이 라이브러리를 받기 위해서는 nvidia 싸이트 디벨로퍼 가입이 필요함.

https://developer.nvidia.com/rdp/cudnn-download 


cudnn-8.0-windows10-x64-v5.1.zip을 받았으면, 압축을 풀어서 cuda 라이브러리가 인스톨된 디렉토리에 복사.

내 경우는 

c:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0>


5) VC++ 재배포가능 패키지 설치

vc_redist.x64.exe

https://www.microsoft.com/ko-kr/download/details.aspx?id=48145


######################


1
2
3
4
5
6
7
8
9
10
11
12
13
>> python -m pip install --upgrade pip
>> conda create -n tensorflow python=3.5
>> activate  tensorflow
conda 환경이 실행되면서, 커서 앞에 (tensorflow)가 붙음.
(tensorflow) >> pip3 install --upgrade tensorflow-gpu    # GPU
(tensorflow) >> pip3 install --upgrade tensorflow    # no GPU. 
 
(tensorflow) >> python
>> import tensorflow as tf    # for test
(tensorflow) >> python main.py
ImportError: No module named 'past'
(tensorflow) >> pip install future
(tensorflow) >> python main.py
cs

끝.


trouble shooting.


#1 python을 찾을 수 없다고 나오면?

제어판 -> 정보 -> 고급시스템 설정 -> 환경변수를 누르면 팝업 메뉴 뜸

아래 쪽에 있는 시스템 변수 탭에서 path 항목 더블클릭.


1
2
3
4
5
c:\Users\eof\Anaconda3
c:\Users\eof\Anaconda3\Scripts
c:\Users\eof\Anaconda3\Library\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp
cs


가 들어있는지 확인. 없으면 추가하고 확인.


#2 python 실행 후에, import tensorflow as tf 명령을 실행하면, ImportError: No module named 'tensorflow' 에러 발생.


에러 화면

1
2
3
4
import tensorflow as tf
Traceback (most recent call last):
File "", line 1in 
ImportError: No module named 'tensorflow'
cs


해결방법.

1
2
3
pip install --upgrade -I setuptools
install TensorFlow again with --ignore-installed flag:
pip install --ignore-installed --upgrade tensorflow-gpu
cs


#3 ImportError: No module named pywrap_tensorflow_internal 에러.


에러 화면 

1
2
3
4
5
ImportError: No module named pywrap_tensorflow_internal
Failed to load the native TensorFlow runtime.
See https://www.tensorflow.org/install/install_sources#common_installation_problems
for some common reasons and solutions.  Include the entire stack trace
above this error message when asking for help.
cs


해결 방법.

1. cuDNN 라이브러리 버전 확인. cuDNN 최신버전 말고, 꼭 버전 5.1(for CUDA 8.0) 이어야함.

2. vc++ 재배포 패키지 설치 유무 확인. 그냥 마음 편하게 다시 깔아도 됨.

https://www.microsoft.com/ko-kr/download/details.aspx?id=48145


끝.