Computer/Ubuntu, Linux
ubuntu 18.04 + Nvidia driver418 + Cuda10.0 환경셋팅
매직블럭
2021. 1. 28. 10:59
최신 그래픽 카드를 쓰면 CUDA 버전이 11 이상으로 올라가겠지만
필요에 따라 CUDA 10.0 환경 세팅이 필요한 경우가 있다.
이런저런 방법들이 인터넷에 있지만 예전에는 쉽게 됐던 방법도
버전이 바뀌고 환경이 바뀌고 하다보니 꼬이고 잘 설치가 안되고 하는 경우가 자주 보인다.
그래서 고생하다 성공한 ubuntu 18.04 + Nvidia driver418 + Cuda10.0 환경셋팅 방법 기록.
(추가) ubuntu 18.04.3 버전 이후로는 커널 버전이 달라 nvidia driver와 문제가 있을수도 있다는 내용을 확인
문제가 심할 경우 ubuntu 18.04LTS 버전으로 설치하여 진행 해 보자..
1. 기존 CUDA 제거
sudo apt-get purge nvidia* && sudo apt-get autoremove && sudo apt-get autoclean && sudo rm -rf /usr/local/cuda*
sudo reboot
2. NVIDIA repository 추가
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt-get update
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt-get update
3. NVIDIA driver 설치
sudo apt-get install --no-install-recommends nvidia-driver-418
sudo reboot
4. CUDA & CuDNN 설치
sudo apt-get install --no-install-recommends cuda-10-0 libcudnn7=7.6.2.24-1+cuda10.0 libcudnn7-dev=7.6.2.24-1+cuda10.0
5. TensorRT 설치 (Optional)
sudo apt-get install -y --no-install-recommends libnvinfer5=5.1.5-1+cuda10.0 libnvinfer-dev=5.1.5-1+cuda10.0