Tensorflow GPU환경 구축을 위한 CUDA Toolkit, cuDNN v5.1 두 가지 프로그램을 설치해야 한다.

1. CUDA Toolkit을 설치한다.

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


2. cuDNN 설치한다.

https://developer.nvidia.com/cudnn


cuDNN의 경우 Nvidia Developer사이트의 회원 가입이 필요하므로 가입하면 된다.

주의 할점은 윈도우 10환경에서는 Download cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0 즉, 5.1버전으로 다운로드 받아야 오류가 발생하지 않는다.


3. 텐서플로우 GPU버전 설치한다.

pip install --upgrade tensorflow-gpu 


4. anaconda prompt에서 방금 만든 환경으로 접속한다.

activate tensorflow


'4차산업 > 머신러닝&딥러닝' 카테고리의 다른 글

Tensorflow GPU 환경구축  (0) 2018.04.03
[링크] 머신러닝,딥러닝  (0) 2018.04.03
머신러닝의 역사  (0) 2018.04.03
블로그 이미지

오픈이지 제로킴

시큐어코딩 교육/컨설팅 전문가 그룹

티스토리 툴바