본문 바로가기

Lab Notes

윈도우10에 Tensorflow GPU 버전 및 Keras 설치

- 최근 새 노트북 구입 후 Keras 설치하고자 함.

- 목표: Tensorflow-gpu 설치, Keras 설치, Jupyter notebook 사용 (Anaconda 기반)

- 윈도우 프롬프트는 관리자 모드로 실행

 

1. Anaconda 설치 (https://www.anaconda.com/distribution/)

    - Python 3.7 version

    - Anaconda Prompt 실행

    - VirtualEnv 설치: pip install virtualenv

    - VirtualEnv 생성: conda create -n venv pip python=3.7

       -- VirtualEnv 진입: conda activate venv

       -- VirtualEnv 해제: conda deactivate

 

2. CUDA 및 CuDNN 설치

     - 현재 기준 (2019.07.14) Tensorflow (>= ver 1.13)은 CUDA 10 만 지원

        -- (https://developer.nvidia.com/cuda-toolkit-archive)

     - CUDA 10 기준 CuDNN 설치

        -- (https://developer.nvidia.com/cudnn)

     - 설치 후 PATH 설정 (윈도우 커맨드 창에 입력)

        1) SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin;%PATH%

        2) SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64;%PATH%

        3) SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include;%PATH%
        4) SET PATH=C:\tools\cuda\bin;%PATH% 

        * 위 4) 의 경우 CuDNN 압축 푼 폴더를 해당 디렉토리에 복사 후 PATH 등록

 

3. Tensorflow-gpu 설치

     - VirtualEnv 진입 상태에서,

     - pip install tensorflow-gpu

 

4. Keras 설치

     - VirtualEnv 진입 상태에서,

     - pip install keras

 

5. 동작 확인

     - (https://keras.io/getting-started/sequential-model-guide/) 소스 코드를 참고하여 설치/동작 상태 확인

 

6. Jupyter notebook 에서 VirtualEnv 사용

     - VirtualEnv 진입 상태에서, 

     - pip install ipykernel

     - python -m ipykernel install --user --name venv --display-name "Keras"

 

 

7. (옵션) 커널 이름 바꾸기

     - C:\Users\사용자 이름\AppData\Roaming\jupyter\kernels\python3 폴더 이동

     - kernel.json 파일을 열고, display_name 뒤의 문자열 변경 

 

 

 

8. (옵션) Tensorflow cpu 버전 추가

     - conda create -n venv-cpu pip python=3.7

     - VirtualEnv 진입: conda activate venv-cpu  (해제:  conda deactivate)

     - tensor flow CPU 버전 설치: pip install --upgrade tensorflow

     - keras 설치: pip install keras

     - Jupyther notebook 설정: pip install ipykernel

     - python -m ipykernel install --user --name venv-cpu --display-name "keras-cpu"

 

9. (참고) 경로 정보

     - VirtualEnv 폴더 경로

        -- C:\Users\사용자 이름\Anaconda3\envs\venv

        -- C:\Users\사용자 이름\Anaconda3\envs\venv-cpu

     - Jupyter kernel 설정 파일 경로 (kernel.json)

        -- C:\Users\사용자 이름\AppData\Roaming\jupyter\kernels\venv

        -- C:\Users\사용자 이름\AppData\Roaming\jupyter\kernels\venv-cpu