종달랩 종다리 이야기

ubuntu 16.04에 tensorflow설치완료

Computer Science/머신 러닝
반응형

구글의 러닝머신 라이브러리 TensorFlow를 우분투에 설치 완료했다.


참고자료 

[TensorFlow 문서 한글 번역본]


GPU 설치참조

[텐서플로우를 파이참에서 돌려보자]


CuDNN 다운로드 : 로그인은 해야 합니다.

[nvidia cuDNN Download]


파이참 설치

[텐서플로우를 파이참에서 돌려보자]


저는 환경을 python3 + GPU 버전으로 했고, CUDA는 7.5, cuDNN은 5.1(August 10, 016) for cuda 7.5로 설정하였습니다.


예제 소스는 python2 버젼이여서 오류가 있어서 잠깐 검색해서 찾아내고

python3에서는 

- xrange가 range로

- print는 print()로 감싸게

변경이 되었더군요.


ps : 오.. 몇주전부터 개발환경에 우분투를 투입시키고 나서 느끼는 것은, 윈도우보다 빠르다는 것과, 좀 있어보인다는 것..


ps2 : GPU가 ignore gpu device...어쩌구 하는 에러가 나면 tensorflow를 직접 컴파일해서 사용할 수 있습니다.

[직접 컴파일 참조링크]

일본어 사이트지만, 요즘은 번역기술이 뛰어나서(여기도 머신러닝을 적용했다지요) 번역해서 읽어도 됩니다. 




반응형