NVIDIA CUDA Toolkit

다운로드

https://developer.nvidia.com/cuda-downloads에서 받을 수 있다. 최신 버전은 7.5이다.

다음과 같이 Target Platform을 선택하고 다운로드 한다.

아래와 같이 받을 수도 있다.

$ curl http://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb

설치

아래와 같이 입력 후, 나머지는 화면에서 Next만 누르면 된다.

$ sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

경로 설정

CUDA 라이브러리와 실행 파일을 위한 경로를 .bashrc 파일에 추가해야 한다.

export CUDA_ROOT=/usr/local/cuda-7.5
export LD_LIBRARY_PATH=${CUDA_ROOT}/lib64:${LD_LIBRARY_PATH}
export PATH=${CUDA_ROOT}/bin:${PATH}

드라이버 설치 확인

아래 명령을 통해 확인할 수 있다.

$ nvidia-smi

오류 메시지가 나올 경우에는 재부팅 후 다시 시도하자.

설치 후 설정

기본적으로 TensorFlow는 CUDA 7.0을 지원하므로, 7.5를 사용하기 위해서는 약간의 trick이 필요하다. 아래의 명령을 이용하여 /usr/local/cuda/lib64의 *.so.7.5에 해당하는 모든 링크를 *.so.7.0으로 만들어준다.

$ for f in *.so.7.5; do sudo ln -s $f ${f%.5}.0; done

# 확인
$ ls -al *.7.?