본문 바로가기
PROGRAMMING/Keras

Windows, Anaconda, Keras 환경설정 (GPU지원이슈)

by BLADEBONE 2018. 7. 11.

윈도우에서 Anaconda 환경에 Keras를 설치 절차 및 이슈에 대한 정리

 

 

◎ 기본환경 : Anaconda 및 CUDA 지원을 위한 드라이버, toolkit 및 cuDNN 설치는 별도로 완료되어 있는 환경 

 

 

설치 절차

① Conda 업데이트

conda update conda

 

② Conda 환경 생성

conda create --name keras

 

참고로 위 명령은 환경이름이 'keras'인 경우로 이름은 자유롭게 지정할 수 있다.

 

◎ Syntax : conda create --name <환경이름>

 

③ 생성한 환경 활성화

activate keras

 

④ Keras 설치 (Backend → TensorFlow)

Keras 설치 시 유의할 사항은 CPU 버전과 GPU 버전을 환경에 맞게 설치하여야 한다.

 

■ CPU 버전

 

conda install keras

 

 

■ GPU 버전

 

conda install keras-gpu

 

Keras 설치 안내에는 backend를 먼저 설치하라고 되어 있으나 conda를 이용하여 keras 설치할 경우 backend로 TensorFlow가 자동으로 설치된다. 주의할 사항은 "conda install keras"로 설치 할 경우 TensorFlow CPU 버전이 설치되기 때문에 신경망 학습을 시켜보면 GPU는 놀고 있고, CPU만 혹사 당하는 안타까운 일이 발생한다. 이를 예방하기 위해 GPU 사용 환경에서는 반드시 "conda insall keras-gpu" 로 설치하여 GPU를 사용할 수 있도록 하는 것이 필요하다.

 

설치 완료 후 python을 실행하고, keras를 import하면 아래와 같은 결과를 얻을 수 있다.

 

 

 

 

반응형

댓글