1. 程式人生 > >ubuntu使用命令安裝GPU驅動和CUDA

ubuntu使用命令安裝GPU驅動和CUDA

1、檢視顯示卡資訊:

lspci | grep -E "VGA|3D" (或者lspci | grep -i nvidia)

2、檢視可以安裝的驅動:

ubuntu-drivers devices

3、假如看到nvidia-361驅動可用,忽略下面的nouveau(可以加入黑名單),接著使用如下命令安裝NVIDIA驅動、N卡的設定面板以及雙顯示卡切換工具:

sudo apt-get install nvidia-361 nvidia-settings nvidia-prime

4、重新啟動:

reboot

5、安裝cuda包:

sudo apt-get install nvidia-cuda-toolkit

當使用ubuntu18.04的時候,安裝的是cuda9.1

6、安裝cudnn:

拷貝一份解壓後的cudnn到宿主機器的/home下,然後如下:

cp -f cuda/include/cudnn.h /usr/local/include/

cp -f cuda/lib64/libcudnn* /usr/local/lib

chmod 777 /usr/local/include/*

chmod 777 /usr/local/lib/*

7、建立軟連線:

sudo ln -sf libcudnn.so.5.10 libcudnn.so.5

sudo ln -sf libcudnn.so.5 libcudnn.so

8、測試:

nvidia-smi

如果出現表格資訊,則驅動安裝成功。

nvcc -V

如果出現cuda版本資訊,則cuda安裝成功。

參考:https://blog.csdn.net/Real_Myth/article/details/52288635?locationNum=8&fps=1