Ubuntu18.04安裝N卡驅動、CUDA10.0、cuDNN7.5.0
N卡驅動
先添加源:
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
然後執行:
ubuntu-drivers devices
查看合適的顯卡驅動,顯示如下:
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001F11sv00001462sd0000126Abc03sc00i00
vendor : NVIDIA Corporation
driver : nvidia-driver-418 - third-party free
driver : nvidia-driver-430 - third-party free recommended
driver : xserver-xorg-video-nouveau - distro free builtin
這裏雖然推薦安裝nvidia-driver-430
,但是親測重啓後無法進入圖形界面,因此選擇安裝nvidia-driver-418
:
sudo apt install nvidia-driver-418
安裝並重啓後,可以打開N卡設置界面,確認驅動安裝成功:
nvidia-settings
CUDA 10.0
先從官網上下載安裝程序,本人選擇的是:
按照官網給的教程:
sudo sh cuda_10.0.130_410.48_linux.run
在安裝的時候,記得看清選項,不用再安裝顯卡驅動了
安裝完後:
sudo gedit ~/.bashrc
在文件最後加上:
export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64”
export CUDA_HOME=/usr/local/cuda
export PATH="$CUDA_HOME/bin:$PATH"
保存修改,並更新配置文件:
source ~/.bashrc
打開新終端,檢查Cuda是否安裝成功:
nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
cuDNN 7.5.0
先從官網上下載,本人選擇的是:
- 假設下載完後解壓縮至
/home/admin/Downloads/CUDA
,那麼在/home/admin/Downloads/CUDA/cudnn-10.0-linux-x64-v7.5.0.56/cuda/
下有include
和lib64
兩個文件夾- Cuda默認安裝位置
/usr/local/cuda-10.0
sudo cp -r /home/admin/Downloads/CUDA/cudnn-10.0-linux-x64-v7.5.0.56/cuda/* /usr/local/cuda-10.0/
檢查cuDNN是否安裝成功
輸入如下命令:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2