소개

2018년 11월 현재 PyTorch가 공식적으로 CUDA 10.0을 지원하지 않으므로 CUDA 9.2를 Ubuntu 16.04에 설치합니다. 이 때, NVIDIA GeForce RTX 20 시리즈를 싣고 있다고 문제가 일어났기 때문에, 해결 방법을 기재.

증상

NVIDIA GeForce RTX 20 시리즈는 Driver 410 이상이 필요. 그러나 Driver 410을 설치한 후 CUDA 9.2를 설치하면 강제로 Driver 396으로 낮아진다. Driver 396에서는 GeForce RTX 20 시리즈에 대응하고 있지 않기 때문에, 로그인 루프에 빠져 버린다.

환경

우분투 16.04.5 LTS (Xenial Xerus)

NVIDIA GeForce RTX 2070

Bash 4.3.48

해결 절차

거친 흐름

드라이버를 설치하지 않고 CUDA 9.2 설치

CUDA와 동시에 자동으로 설치된 드라이버 제거

PPA에서 드라이버 410 설치

자세한 단계

1. 소개 NVIDIA Driver는 설치하지 않습니다.

만약 설치하고 있는 경우는 다음과 같이 하여 삭제한다.

terminal

$ sudo apt --purge remove nvidia-*  # Driverの削除
$ sudo apt --purge remove cuda-*  # CUDAの削除
$ sudo apt autoremove  # 不要になったファイルの削除

또한, 로그인 루프에 빠져있을 때는 ctrl + alt + F1로 CUI로 로그인 한 후 위의 명령을 실행합니다.
ctrl + alt + F7로 GUI로 돌아갈 수 있습니다.

2. CUDA 9.2 다운로드

CUDA 9.2를 아래 사이트에서 다운로드
htps : //로 ゔぇぺぺr. 응아아. 코 m / 쿠다 - 92 - 드 w ぉ 아 드 아 r ゔ ぇ
이 때, Installer Type은 「deb(local)」를 선택.

3. CUDA 9.2 설치

(예)

terminal

$ cd ~/Downloads
$ sudo dpkg -i cuda-repo-ubuntu1604-9-2-local_9.2.148-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub
$ sudo apt update
$ sudo apt install cuda

4. CUDA 설치와 동시에 자동으로 설치된 드라이버 제거

terminal

$ dpkg -l | grep nvidia-*  # これでインストール済みのパッケージが表示される。
$ sudo apt --purge remove nvidia-*  # Driverの削除
$ dpkg -l | grep nvidia-*  # これで何も表示されなければ削除成功

5. CUDA 경로 설정

.bashrc 등에 다음을 추가

.bashrc 등에 추가

# CUDA
export PATH=/usr/local/cuda-9.2/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64:${LD_LIBRARY_PATH}

6. PPA 추가

terminal

$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update

7. Driver 410 설치

terminal

$ sudo apt install nvidia-410

8. 재부팅

terminal

$ reboot