윈도우 Tensorflow 환경 구축하기

[2020년 12월 9일 기준 수정(tensorflow 2.3.1)]

윈도우 10에 Tensorflow 설치하는 방법을 정리해 보았습니다.

먼저, TensorFlow를 설치함에 있어, 공식문서만큼 중요한 참고 문서가 없을 것입니다.

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

아래 제가 진행한 과정은 참고만 하시고, 공식문서를 바탕으로 설치하세요!!

제 준비물은 다음과 같습니다.

- 운영체재 : 윈도우 10

- 그래픽카드 : Geforce GTX 1060 6GB

- NVIDIA Driver : 430.86-desktop-win10-64bit-international-whql

- Anaconda Version : Anaconda3-2020.07-Windows-x86_64 (Python 3.8.3) 

- CUDA : cuda_10.1.105_418.96_win10 (link)

- CUDNN :cudnn-10.0-windows10-x64-v7.6.1.34 (link)

※ 이 때, cuda와 cuDNN 의 버전을 잘 확인해야 합니다. CUDA 10.1과 호환되는 cuDNN 버전은 v8.0.5 입니다.

작업 순서

(0. NVIDIA Driver 설치)

1. Anaconda 설치

2. CUDA 설치

3. CUDNN 설치 : 다운받은 파일을 압축해제 후 bin, include, lib 폴더를 CUDA 설치 경로에 복사 붙여넣기 

   C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

4. Tensorflow 패키지 설치

Anaconda 가상환경을 만들어 설치합니다. 혹시 "엑세스가 거부되었습니다." 라는 에러가 발생하면, CMD 창을 "관리자권한으로 실행"합니다.

$ conda create -n tf    -> tf 라는 이름의 가상환경 생성
$ activate tf           -> 활성화
$ (tf) pip install tensorflow       -> tensorflow 설치

그리고, 터미널 창을 껐다가 다시 수행해 봅니다.

$ activate tf

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

5. (선택) Jupyter notebook 원격 접속하기

cmd 창에서 아래 명령어 실행합니다.

$ jupyter notebook --generate-config

Writing default config to: C:\Users\user\.jupyter\jupyter_notebook_config.py

라는 결과가 나온다. 해당 경로에 가면 jupyter_notebook_config.py 라는 파일이 생성됨을 확인할 수 있다.

해당 경로를 잘 기억해 두고 있다가,

python 을 실행하고(그냥 cmd 창에서 python 입력)

다음과 같은 코드를 입력합니다.

In [1]: from notebook.auth import passwd

In [2]: passwd()
Enter password: 
Verify password: 

암호화된 비밀번호가 생성되는데,

이를 잘 복사해 두었다가,

jupyter_notebook_config.py 를 실행합니다.

그러면, 다양한 option 들이 주석처리 되어 있는 것을 확인할 수 있습니다. 이 중

# c.NotebookApp.password = ''

c.NotebookApp.password = 'sha1:dd6357b6f9ab:fa31a5edf1790e0a09ca56c44a821bebca57e393'

과 같이 수정.

그리고, 

# c.NotebookApp.ip = 'localhost' 부분도 본인의 ip로 수정하면 됩니다.

* offline 에서 tensorflow-cpu 버전을 설치하고 싶으신 경우

https://www.lfd.uci.edu/~gohlke/pythonlibs 에서 Ctrl+F를 이용, tensorflow를 검색한 후

.whl 파일을 받으신 후,

다음과 같이 pip 명령어를 통해 설치합니다.

pip.exe install --upgrade --no-deps <tensorflow whl file name>

----

[2020.03.01] 수정사항

tensorflow 2.0 버전으로 업데이트하기

pip install --upgrade tensorflow-gpu==2.0.0  #2.1.0 버전은 cuda / cudnn 버전을 업그레이드 해야 합니다.