우분투 16.04와 18.04에서 NVIDIA 그래픽 카드 드라이버를 설치하는 방법을 소개하는 유용한 링크는 다음과 같다.
우분투 버전별로 그래픽 드라이버를 설치하는 방법이 다른 듯하다.
1. 우분투 18.04 버전
18.04버전에서 그래픽 드라이버를 설치하기 위해 아래 링크를 참고하면 좋다.
https://codechacha.com/ko/install-nvidia-driver-ubuntu/
Ubuntu18.04 LTS에 nvidia 드라이버를 설치하는 방법 | codechacha
우분투 18.04를 설치 후 화면이 멈추는 일이(lockup) 잦아졌습니다. 원인은 nvidia 그래픽카드 드라이버를 설치하지 않아 발생한 문제였습니다. 드라이버는 수동, 자동으로 설치하는 방법이 있습니다. 먼저 자신이 설치해야 하는 그래픽 드라이버 버전을 알아야 하고 드라이버를 설치해야 합니다.
codechacha.com
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update
$ apt-cache search nvidia | grep nvidia-driver-418
nvidia-driver-418 - NVIDIA driver metapackage
$ sudo apt-get install nvidia-driver-418
$ sudo reboot간단하게 위의 명령어들을 입력해 주면 된다.
ppa를 이용하는 방법이 간단하고 에러없이 잘 동작한다.
2. ubuntu 16.04 버전에서 엔비디아 그래픽 드라이버 설치
https://www.tensorflow.org/install/gpu
GPU 지원 | TensorFlow
TensorFlow GPU 지원에는 다양한 드라이버와 라이브러리가 필요합니다. 설치를 단순화하고 라이브러리 충돌을 방지하려면 GPU를 지원하는 TensorFlow Docker 이미지를 사용하는 것이 좋습니다(Linux만 해당). 이 설정에는 NVIDIA® GPU 드라이버만 있으면 됩니다. 아래의 설치 명령어는 최신 버전의 TensorFlow에 적용됩니다. 이전 TensorFlow 출시에 사용할 CUDA 및 cuDNN 버전은 테스트된 빌드 구성을 참조하세요
www.tensorflow.org
텐서플로우 GPU 버전 설치 가이드를 따라하면 쉽게 설치할 수 있다.
# Add NVIDIA package repositories
# Add HTTPS support for apt-key
$ sudo apt-get install gnupg-curl
$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_10.0.130-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1604_10.0.130-1_amd64.deb
$ sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ sudo apt-get update
$ wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb
$ sudo apt install ./nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb
$ sudo apt-get update
# Install NVIDIA driver
# Issue with driver install requires creating /usr/lib/nvidia
$ sudo mkdir /usr/lib/nvidia
$ sudo apt-get install --no-install-recommends nvidia-410
# Reboot. Check that GPUs are visible using the command: nvidia-smi
# Install development and runtime libraries (~4GB)
$ sudo apt-get install --no-install-recommends \
cuda-10-0 \
libcudnn7=7.4.1.5-1+cuda10.0 \
libcudnn7-dev=7.4.1.5-1+cuda10.0
# Install TensorRT. Requires that libcudnn7 is installed above.
$ sudo apt-get update && \
sudo apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda10.0 \
&& sudo apt-get update \
&& sudo apt-get install -y --no-install-recommends libnvinfer-dev=5.0.2-1+cuda10.0
16.04버전에 대해 위의 명령어를 입력하면 따로 blacklist 설정 등을 할 필요가 없이 쉽게 설치 가능하다.
텐서플로우 홈페이지를 보면 18.04버전에 대한 명령어도 나와있다.
댓글