딥러닝 학습을 위해 우분투 CUDA 버전 바꾸기

pytorch나 tensorflow를 이용하다보면 버전에 따라 맞는 CUDA 버전도 같이 바꾸어 줄 필요가 있다. 바꾸는 방법은 굉장히 간단하다.

1. CUDA 설치

본 블로그에 "CUDA 및 cuDNN 설치 및 제거방법"의 설치를 따라 CUDA를 설치해준다.
(https://jonghoonp.blogspot.com/2019/08/cuda-cudnn.html)

2. 설치된 CUDA 버전 확인

자신이 이용할 버전의 폴더를 아래 명령어로 확인한다.

cd /usr/local
ls

예시에서는 CUDA 10 과 8버전 디렉토리가 있는 것을 확인 할 수 있다.

3. PATH 설정

설치한 CUDA 버전 중 사용할 버전의 디렉토리 PATH 설정을 하면 된다.

vi ~/.bashrc

그리고 아래 그림처럼 PATH부분을 찾아 버전부분만 수정한다. 없다면 직접 입력해주면 된다.

위 그림을 아래처럼 빨간색 이름 부분에 version만 바꾸어준다 ( 8 -> 10 버전으로 바꿈)

exportPATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}} 
export LD_LIBRARY_PATH=/usr/local/cuda-10.0

        /lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

4. source ~/.bashrc

Path 적용을 위해 vi에서 빠져나온 후 아래명령어를 실행 한다.

source ~/.bashrc

실행 후 원하는 버전이 제대로 적용되었는지 확인 해보면 된다.
nvcc --version

cuDNN 적용을 하고자 한다면 위 1.번의 CUDA 설치 포스트에서 나와있는 방법대로 하면 된다.

딥러닝 학습을 위해 우분투 CUDA 버전 바꾸기