[Tensorflow] 아나콘다 가상환경에서 텐서플로우 설치하기

2019년 말, 텐서플로우 2.0 버전이 배포되면서 머신러닝 분야에서 텐서플로우의 열기는 더욱 더 뜨거워졌습니다. 새로워진 텐서플로우 설치를 시작으로 CNN (Convolutional Neural Network) 기반의 이미지 분류기, Object detection 알고리즘 중 하나인 Faster R-CNN까지 시리즈로 포스팅을 이어나가도록 하겠습니다.

텐서플로우 CPU 모드의 경우 설치가 간편하지만, GPU 모드로 구동하실 경우 파이썬이나 CUDA 버전에 의존성이 강하여 초기 설치환경 구축이 매우 중요합니다. 설치되는 라이브러리 또는 프로그램의 모든 버전을 확인하시고 설치해주시기 바랍니다.

1. 가상환경 생성

// 가상환경 생성

conda create -n py38tensorflow python=3.8

// 가상환경 활성화

activate py38tensorflow                                       

원활한 구동을 위해 순정 상태의 가상 환경을 구축하였습니다. 작성 날짜 기준으로 최신 버전인 텐서플로우 2.2 설치를 위해 파이썬 3.8을 기반으로 생성합니다.

2. 텐서플로우 설치

pip install tensorflow==2.2

텐서플로우 2.2를 pip로 설치해줍니다. 용량이 상당히 커서 인터넷 환경에 따라 시간이 다소 소요될 수 있습니다. 가끔 설치중 오류로 중단되는데, 당황하지 마시고 다시 설치해주시면 됩니다.

3. 비주얼스튜디오 재배포 패키지 설치

다운로드 : https://support.microsoft.com/ko-kr/help/2977003/the-latest-supported-visual-c-downloads

비주얼스튜디오 재배포 패키지 2015를 설치해줍니다. 자신의 윈도우 환경에 맞는 버전을 설치해주시면 됩니다.

4. CUDA 설치

// CUDA 10.1 version

다운로드 : https://developer.nvidia.com/cuda-10.1-download-archive-update2

// cuDNN 7.6.5 version

다운로드 : https://developer.nvidia.com/rdp/cudnn-download

텐서플로우 2.2 기준으로 무조건 CUDA 10.1이 요구됩니다! 10.2는 안됩니다... 링크를 통하여 CUDA 10.1을 우선 설치해줍니다. 설치가 진행되는 동안 cuDNN을 다운받으시면 됩니다. cuDNN은 로그인 이후 다운로드 할 수 있으니, 회원이 아니시라면 가입을 먼저 해줍니다.

Download cuDNN 클릭 후 로그인을 해줍니다.

이후, cuDNN 7.6.5를 다운로드 해주시면됩니다. Download cuDNN v7.6.5 [November 5th, 2019], for CUDA 10.1 입니다!!!

CUDA 설치가 완료되면 cuDNN을 CUDA가 설치된 경로에 압축을 풀어주신 뒤, 복사해주시면 됩니다. 일반적인 경로는 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1 입니다.

 

5. 설치 확인

import tensorflow as tf

tf.__version__

텐서플로우 기본 환경 테스트를 위해 파이썬을 실행한 다음 텐서플로우를 import 해줍니다. '2.2.0' 이 뜨시면 설치완료입니다.

# 머신러닝 프로젝트 제작, 상담 및 컨설팅  / 머신러닝 접목 졸업작품 컨설팅

# 데이터 가공, 수집, 라벨링 작업 / C, 파이썬 프로그램 제작

# email : deepi.contact.us@gmail.com

# site : www.deep-i.net