개발/삽질 기록
conda tensorflow 설치 및 jupyter notebook 연결
kwony
2018. 8. 3. 23:20
Conda 환경에서 tensorflow를 설치하는 방법
0. Prerequsite
- Anaconda를 설치한다.
- GPU를 사용할 경우 tensorflow를 사용할 수 있도록 CUDA, cuDNN 라이브러리를 설치한다.
1. tensorflow 가상 환경 생성
tensorflow를 사용하는 가상 환경을 하나 생성한다. 별도로 설정하고 싶은 환경이 있으면 생략하고 아니면 만들자.
kwony@kwony:~$ conda create -n tensorflow pip \ python={ python 버전 입력 ex) 2.7, 3.4
2. 가상환경 활성화
1에서 만들어둔 가상 환경을 활성화한다. activate 명령어가 안먹히는 경우는 anaconda의 바이너리가 export가 제대로 안된것이니 확인해보도록 하자.
kwony@kwony:~$ conda activate tensorflow (tensorflow)$ # tensorflow로 명령어 프롬프트가 변형된다
3. tensorflow 설치
활성화된 명령어 프롬프트에서 tensorflow를 설치한다. tensorflow 공식 홈페이지에서 설치하려는 tensorflow의 종류와 파이선 버전에 해당하는 저장소를 입력하도록 하자.
(tensorflow) kwony@kwony:~$ pip install --ignore-installed \ --upgrade { URL }
4. jupyter 설치
원래 아나콘다를 설치할 때 기본으로 포함된 jupyter-notebook에서 tensorflow를 쓰려고 하면 없는 모듈이라고 나온다. 방금전에 설치한 tensorflow는 현재 사용중인 가상 환경에만 적용되기 때문이다. 현재 가상 환경에서 사용하는 jupyter notebook을 설치하면 방금 전에 설치한 tensorflow를 사용 할 수 있다.
(tensorflow) kwony@kwony:~$ pip install jupyter
5. jupyter notebook 실행 후 테스트
jupyter 노트북으로 tensorflow 모듈을 사용해보면 문제없이 동작하는 것을 확인 할 수 있다.
(tensorflow) kwony@kwony:~$ jupyter-notebook