[Linux] Keras Backend 변경

Keras를 사용하다보면 keras using theano backend.,  keras using tensorflow backend. 등 자신이 원치않는 프레임 워크를 불러오는 경우가 있다.

 

이럴 경우 자신이 원하는 프레임 워크를 명시적으로 지정해줘야한다.

 

Ubuntu 16.04 / Anaconda3를 기준으로 Keras Backend를 Tensorflow로 바꾸는 예시이다.

 

$ source activate YOUR_ENV
(YOUR_ENV)$ set "KERAS_BACKEND=tensorflow"

 

다시 Theano로 바꾸려면

(YOUR_ENV)$ set "KERAS_BACKEND=theano"

 

매번 바꾸기 귀찮을 경우 ~/.bashrc에 지정해두자.

 

 

 


# 위 방법대로 했으나 바뀌지 않을 경우

$ which conda
/home/{USER}/anaconda3/bin/conda

$ cd /home/{USER}/anaconda3/envs/{YOUR_ENV}/etc/conda
$ ls
activated.d  deactivated.d

activated.d 는 Default Keras backend를 정의

deactivated.d는 Default Keras backend를 해제

 

권장하진 않으나, 아래의 방법을 통해 임시방편으로 해결할 수 있음.

 

$ vi activate.d/keras_activate.sh

:wq

$ vi deactivate.d/keras_deactivate.sh

:wq

 

댓글 남기기