Ubuntu 18.04에 CUDA 10.2

https://whiteglass.tistory.com/15

터미널에 $ nvcc -V 혹은 $ nvidia-smi 치면 CUDA 버전 확인할 수 있음.

내 노트북엔 11.4가 깔려있는데 10.2가 필요해서 다운그레이드

1) 기존 CUDA 삭제 후 재부팅

$ sudo apt-get --purge remove "*cublas*" "cuda*"
$ sudo reboot

근데 제대로 삭제되지 않고 아래와 같이 나옴

패키지 목록을 읽는 중입니다... 완료
의존성 트리를 만드는 중입니다       
상태 정보를 읽는 중입니다... 완료
Note, selecting 'libcublas9.1' for glob '*cublas*'
Note, selecting 'cuda-repo-ubuntu1204' for glob 'cuda*'
Note, selecting 'cuda-repo-ubuntu1404' for glob 'cuda*'
Note, selecting 'cuda-repo-ubuntu1410' for glob 'cuda*'
Note, selecting 'cuda-repo-ubuntu1504' for glob 'cuda*'
Note, selecting 'cuda-license-7-5' for glob 'cuda*'
Note, selecting 'cuda-repo-ubuntu1404-7-5-local' for glob 'cuda*'
Note, selecting 'cuda-drivers-fabricmanager-450' for glob 'cuda*'
Note, selecting 'cuda-drivers-fabricmanager-460' for glob 'cuda*'
Note, selecting 'cuda-drivers-fabricmanager-470' for glob 'cuda*'
Note, selecting 'cuda-repo-ubuntu1504-7-5-local' for glob 'cuda*'
Note, selecting 'cuda-drivers-fabricmanager' for glob 'cuda*'
패키지 'cuda-license-7-5'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'cuda-repo-ubuntu1204'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'cuda-repo-ubuntu1404'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'cuda-repo-ubuntu1404-7-5-local'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'cuda-repo-ubuntu1410'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'cuda-repo-ubuntu1504'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'cuda-repo-ubuntu1504-7-5-local'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'libcublas9.1'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'cuda-drivers-fabricmanager-450'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'cuda-drivers-fabricmanager-460'는 설치되어 있지 않아, 지우지 않았습니다.
패키지 'cuda-drivers-fabricmanager-470'는 설치되어 있지 않아, 지우지 않았습니다.
다음 패키지가 자동으로 설치되었지만 더 이상 필요하지 않습니다:
  linux-objects-nvidia-510-5.4.0-99-generic
  linux-signatures-nvidia-5.4.0-99-generic
Use 'sudo apt autoremove' to remove them.
0개 업그레이드, 0개 새로 설치, 0개 제거 및 2개 업그레이드 안 함.

그래서 sudo apt autoremove 해줌

2) CUDA Toolkit 홈페이지에서 원하는 버전을 찾기

https://developer.nvidia.com/cuda-toolkit-archive

나는 10.2 찾아서 아래 사진과 같이 선택함

3) 설치 과정에 필요한 명령어 입력

위 사진처럼 선택하면 아래에 설치를 위한 명령어가 뜸

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
$ sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
$ sudo apt-get update

네 번째 줄 실행시키니까

Selecting previously unselected package cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01.
(데이터베이스 읽는중 ...현재 253409개의 파일과 디렉터리가 설치되어 있습니다.)
Preparing to unpack cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb ...
Unpacking cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01 (1.0-1) ...
cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01 (1.0-1) 설정하는 중입니다 ...

The public CUDA GPG key does not appear to be installed.
To install the key, run this command:
sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub

이렇게 떠서 $ sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub 해줌

4) CUDA 10.2 설치
$ sudo apt-get -y install cuda-10-2

5) 환경 변수 설정
$ sudo vi ~/.bashrc

가장 아래에 다음을 추가

export PATH=$PATH:/usr/local/cuda-10.2/bin
export CUDADIR=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64

6) 환경 변수 불러오기
$ source ~/.bashrc

7) 버전 확인해보기
$ nvidia-smi 입력하면

CUDA 10.2로 잘 깔려있는 걸 볼 수 있다!

추가) $ nvidia-smi를 입력했을때 "Failed to initialize NVML: Driver/library version mismatch"라는 에러가 떴다. 재부팅해주면 된다.