[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
반응형
'AI Development > TensorFlow | TFLite' 카테고리의 다른 글
[TensorFlow] 함수 내부에서 TensorFlow Graph 실행하기 (0) | 2019.08.03 |
---|---|
[TensorFlow] pb 파일 TensorBoard에 띄우기 (TF 1.x 버전용) (0) | 2019.05.31 |
[Tensorflow] Tensorflow GPU 버전 설치하기 (2) | 2018.01.11 |
[Tensorflow] windows 환경에서 tensorflow에 opencv 설치하기 (0) | 2017.09.05 |
[Tensorflow] Windows 환경에서의 Anaconda를 이용한 Tensorflow CPU 버전 설치하기 (에러해결법) (0) | 2017.09.05 |