Computer/Ubuntu, Linux

Ubuntu에 CUDA 설치하기.

매직블럭 2015. 12. 23. 10:08
이전글

머신러닝 공부하려고 우분투 깔았으면 이제 쿠다를 설치 해야지.


이런저런 방법들이 소개 되어 있지만 가장 간단한 설치 성공한 방법으로 정리 해두자.


========================================================================================


1. NVIDIA 홈페이지에서 해당버전 deb 파일 다운로드

https://developer.nvidia.com/cuda-downloads


2. 터미널에서 디패키징 후 쿠다 설치해주기

$ sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb

$ sudo apt-get update

$ sudo apt-get install cuda


3. CUDA 설치 후 nvcc 버전을 확인 해보면 툴킷이 깔리지 않은 것을 확인 할 수 있다.

$ nvcc --version 


4. 툴킷 설치가능하다는 메세지가 뜨지만 해당 명령어는 구버전으로 인도하기 때문에 각자 본인 버전에 맞게 설치하자

$ sudo apt-get install cuda-toolkit-7.5

PATH 설정 해 주면 연결 됨. 5번으로..


5. Toolkit 까지 설치 하였다면 환경변수 설정이 필요하다. 단순히 export 를 이용하여 추가 할 경우 재부팅을 하면 설정한 경로가 해제된다고 한다. bashrc 파일에 추가 해 주는 방식으로 영구 추가 해 주자.

$ vim .bashrc


-- 추가 --

export CUDA_HOME=/usr/local/cuda

export LD_LIBRARY_PATH=${CUDA_HOME}/lib64 

 

PATH=${CUDA_HOME}/bin:${PATH} 

export PATH 

저장후 종료


# update

$ source .bashrc


(optional) 기본 제공하는 vi 에디터는 커서 위치 이동 등이 불편하더라.. 이와 유사한 vim 에디터를 사용하면 조금 더 편리. 

사용법은 vi 에디터와 동일 한듯,? ( vi 에디터 명령어 정리 - vi / vim 에디터 편집을 위한 명령어 정리. )


위 단계를 차례로 수행하면 우분투 환경에서 CUDA를 정상적으로 설치 완료하고 이용 가능하다.