머신러닝 준비 2/4: CUDA 및 cuDNN 설치

CUDA는 GPU가속을 이용한 머신러닝을 도와주는 툴킷이며

cuDNN은 함께 설치 시 CUDA를 이용한 학습 속도가 더 빨라진다.

NVIDIA 홈페이지에서 cuda toolkit 8.0 (Linux > x86_64 > Ubuntu > deb (local))과

cudnn 5 (cuDNN v5 Library For Linux) 를 다운로드 한다.

cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb cudnn-8.0-linux-x64-v5.0-ga.tgz

cuda toolkit을 설치한다.

$ sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb $ sudo apt-get update $ sudo apt-get install cuda

cudnn의 압축을 푼다.

$ tar -zxf cudnn-8.0-linux-x64-v5.0-ga.tgz

압축을 풀면 cuda 디렉토리가 생성된다.

cuda toolkit이 설치된 디렉토리에 cudnn 파일들을 복사한다.

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ sudo cp -a cuda/lib64/* /usr/local/cuda/lib64/

라이브러리 path를 추가하기 위해 .bashrc를 수정한다.

$ sudo vi ~/.bashrc 또는 $ sudo nano ~/.bashrc

.bashrc에 아래 내용을 추가한다.

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64 export CUDA_HOME=/usr/local/cuda

.bashrc에 추가한 내용을 적용한다.

$ source ~/.bashrc