[keras] tensorflow-gpu ImportError 해결 및 keras 간단 설치

xxxxxxxxxx
pip install tensorflow-gpu

위 명령어를 통해 tensorflow gpu를 설치하고 import를 하면 다음과 같은 오류가 날 때가 있다.

xxxxxxxxxx
ImportError: DLL load failed: The specified module could not be found.

일반적으로 CUDA를 설치해야 하기 때문에 발생하는 에러일 가능성이 높다. 인터넷 상에 CUDA 홈페이지에서 이것저것을 다운받아 설치하는 방법들이 많이 소개되어있지만, 최근 keras까지 anaconda를 이용해서 간단히 설치할 수 있는 방법을 소개하고자 한다.

 

방법은 간단하다. 그저 다음 명령어들을 따라 입력하면 된다.

conda install -c aaronzs tensorflow-gpu
conda install -c anaconda cudatoolkit
conda install -c anaconda cudnn
conda install keras-gpu

가상환경을 만들지 않아도 되지만, 가상환경을 만들고 그 안에서 설치하는 것을 추천한다.

 

가상환경 만들기

conda create --name [가상환경명]

 

참고 : https://github.com/tensorflow/tensorflow/issues/22794