Unix/Arch Linux

Keras + Theano 시도: 1. Theano 설치

ForceCore 2016. 5. 11. 10:01
Keras가 Deep learning을 도구화 해서 그냥 갖다 쓰면 된다! 고 해서 써봤다. 생각보다 쉽진 않다. -_-;; 처음 해서 그런지 모든게 낯설다.


우선 Theano가 Cuda 써서 빠르게 도는 것 부터.


이건 그리 어렵지 않다. Arch Linux에서는

extra/nvidia 364.19-3 [installed: 364.19-1]
    NVIDIA drivers for linux

community/cuda 7.5.18-2 [installed: 7.5.18-1]
    NVIDIA's GPU programming toolkit
community/pycuda-headers 2016.1-1 [installed]
    Python wrapper for Nvidia CUDA
community/python-pycuda 2016.1-1 [installed]
    Python wrapper for Nvidia CUDA
community/python2-pycuda 2016.1-1 [installed]
    Python wrapper for Nvidia CUDA

이런 패키지들이 있어서 어렵지 않다.


pip install Theano

로 설치도 해주면 되는데... 아니다. AUR에 있다... ㅋ

yaourt -S python-theano


Environment Variable도 지정해줘야 함.

export CUDA_ROOT=/opt/cuda
이게 하이라이트. ~/.bashrc 에다가 넣으면 된다. 이게 없으면 cuda를 인식 못 해서 CPU로 연산을 시도하기 때문에 모든것이 매~~~우 느리게 된다.


흐 음 ㅋ 압축적으로 적었지만 이 부분도 초보는 힘들 것이다. 공식사이트에 나온 설명도 위에 적은 것과 별반 다르진 않다. 그래서 힘든데... vi 사용법까지 가르쳐줘야 하나... 하는 그런 문제도 있고 -0-;;; 모르면 그건 알아서 찾고.

$ vi ~/.bashrc

자신이 bash를 쓴다면 홈디렉토리 안의 .bashrc 를 수정해야 한다. 로그인 하면 읽는 파일이 저것이다. 맨 마지막에 export CUDA_ROOT=/opt/cuda 한 줄을 넣어준다. 그리고 로그인을 다시 해야지 적용이 된다.


굵직한 것만 적었는데, 사실 자세한 사항이나 주의점은 Theano를 설치하면서 메모하지 않아서 잘 기억 안 난다. 그만큼 설치에 별 신경을 안 써도 잘 되었던듯.


그리고 테스트 파이썬 스크립트를 돌려보자.

import theano
theano.test()

이렇다는데... 왠지 지금 해보니까 안 돈다. nose 패키지가 없다나 뭐라나. 귀찮... 테스트 패스.