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
반응형