AI Development/TensorFlow | TFLite

[TensorFlow] Anaconda 가상환경 이용하여 TensorFlow GPU 설치

꾸준희 2019. 3. 5. 20:50
728x90
반응형




아나콘다 설치 후 


1. 가상환경 생성

conda create -n 가상환경이름 python=3.6 


2. 가상환경 Activation

source activate 가상환경이름


3. TensorFlow GPU 설치

conda install -c anaconda tensorflow-gpu 


(2019년 3월 5일 기준 )이렇게 설치하면 현재 아나콘다 패키지에 나와있는 최신 버전인 TensorFlow 1.12 버전으로 설치되며

CUDA 9.2, cuDNN 7.2.1 로 설정된다.

필자는 CUDA 10.0 버전이 설치된 상태에서 가상환경을 위와 같이 설정하였다. 


텐서플로우 GPU 버전 패키지 확인 https://anaconda.org/anaconda/tensorflow-gpu/files



3-1. TensorFlow GPU 버전 설정하여 설치 시 

conda install -c anaconda tensorflow-gpu=1.x

conda install -c anaconda cudnn=7.x.x or ... 

conda install -c anaconda cudatoolkit=9.x or 10.x ... 


(필요시) conda install -c menpo opencv





참고자료 1 : https://shakeratos.tistory.com/m/9?category=715017

참고자료 2 : https://gldmg.tistory.com/73

728x90
반응형