Tensorflow-gpu 설치 및 Jupyter에 커널 등록하기 on Windows

Tensorflow-gpu 설치 및 Jupyter 등록 on Windows

• Windows, Cuda 9.0, cudnn 7.4, Anaconda는 설치되어 있는 환경에서 진행합니다.
  • Cuda 9.0 : https://developer.nvidia.com/cuda-90-download-archive
    • Cuda 9.0 설치시 Visual Studio Integration인가 한 메뉴가 있는데 이걸 체크했을때 설치가 잘 안된 기억이 있어서 해제했습니다.
    • Cuda쪽만 체크하고 설치해도 무방한듯 합니다.
  • cudnn : https://developer.nvidia.com/rdp/cudnn-download
• Anacomda prompt를 실행합니다.
• 새 가상환경을 만들기 위해 아래를 입력합니다. tf는 가상환경의 이름이고 python version은 3.6으로 했습니다.
  • conda create -n tf pip python=3.6
• 가상환경을 활성화시킵니다.
  • activate tf or source activate tf
• pip를 업그레이드 시킵니다.
  • python -m pip install --upgrade pip
• tensorflow-gpu를 설치합니다.
  • pip install --ignore-installed --upgrade tensorflow-gpu
• Jupyter에 등록하기 위해 아래 명령을 입력합니다.
  • conda install notebook ipykernel
  • python -m ipykernel install --user --name tf --display-name "Tensorflow"
• 환경 확인 : conda info --envs
• conda 환경 삭제 : conda remove --name ENV_NAME --all
• Jupyter에서 커널 삭제 : jupyter kernelspec uninstall tf