0. 컴퓨터 사양 확인

  • 설치 전, 자신의 컴퓨터의 운영체제, GPU를 먼저 확인

제 컴퓨터 사양

  • 운영체제 : Windows 10
  • GPU : Quadro P600

1. NVIDIA driver 설치

NVIDIA driver Download : https://www.nvidia.co.kr/Download/index.aspx?lang=kr

  • 자신의 GPU에 맞는 드라이버 설치

2. CUDA 설치

CUDA Toolkit Archieve : https://developer.nvidia.com/cuda-downloads

  • 운영체제와 버전을 선택하여 설치파일 다운로드(여기서는 CUDA 10.0 설치)
  • 설치파일 실행하여 CUDA 설치

3. cuDNN 설치

cuDNN Download : https://developer.nvidia.com/cudnn

  • 회원가입 필요
  • CUDA에 맞는 버전 설치파일 다운로드(여기서는 cuDNN 10.0 설치)
  • 다운받은 zip 파일을 압축 해제(각 폴더마다 1개의 파일이 들어있음)
  • 압축 해제된 파일들을 CUDA 설치 경로(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA)에 덮어쓰기

4. 가상환경 생성 및 jupyter kernel 생성

  • cmd 관리자 권한으로 실행

  • 가상환경 생성

    conda create --name tf-gpu python=3.7

  • 가상환경 활성화

    activate tf-gpu

  • jupyter kernel 생성

    conda install ipykernel
python -m ipykernel install --user --name tf-gpu --display-name tf-gpu

5. Tensorflow 2.0 gpu 설치

  • 가상환경 활성화된 상태에서 tensorflow 2.0 gpu 버전 설치
conda install tensorflow-gpu==2.0.0

6. 설치 확인

import tensorflow as tf 
print(tf.__version__) 
tf.test.is_gpu_available()

2.0.0
True

Reference