(一)Nvidia驅動安裝
ubuntu 16.04默認已安裝第三方驅動程序nouveau,首次安裝nvidia顯卡驅動首先需要禁用nouveau,不然會碰到沖突的問題,導致無法安裝nvidia顯卡驅動。
1、打開blacklist.conf
sudo vim /etc/modprobe.d/blacklist.conf
(若目錄下沒有blacklist.conf,則創建該文件)
2、在blacklist.conf文件最后部分插入以下兩行內容
blacklist nouveau options nouveau modeset=0
3、更新系統
sudo update-initramfs -u
4、重啟系統(一定要重啟)
sudo reboot
5、驗證nouveau是否已禁用
lsmod | grep nouveau
沒有任何輸出則禁止成功
6、開始安裝驅動
方案1:與cuda一起安裝(推薦)下載路徑:https://developer.nvidia.com/cuda-downloads
關閉圖形化界面(不然會報"You appear to be running an X server"錯誤)
sudo service lightdm stop
安裝命令(這個run文件可以cuda和NVIDIA驅動一起安裝)
sudo sh cuda_10.0.130_410.48_linux.run
安裝成功后,會顯示installed,否則會顯示failed
方案2:當然也可以分開安裝(不推薦,因為驅動版本與cuda版本有關系)
首先下載驅動文件,然后
sudo chmod 777 NVIDIA-Linux-x86_64-418.43.run
安裝NVIDIA驅動(注意參數)
sudo ./NVIDIA-Linux-x86_64-375.20.run –no-opengl-files
–no-opengl-files 只安裝驅動文件,不安裝OpenGL文件。這個參數最重要
–no-x-check 安裝驅動時不檢查X服務
–no-nouveau-check 安裝驅動時不檢查nouveau
后面兩個參數可不加;
安裝完成后重啟即可。
7、安裝完后
sudo service lightdm start
sudo reboot
(二)Nvidia驅動卸載
首先ctrl+Alt+F2/F1進入命令行狀態,然后停止lightdm
sudo service lightdm stop
卸載命令位置/usr/bin/nvidia-uninstall,以下命令即可卸載。
sudo /usr/bin/nvidia-uninstall
不找這個命令的位置,也可以
sudo apt-get install remove --purge nvidia*
卸載完后輸入nvidia-smi命令沒有任何反應即卸載成功。重啟電腦可能會出現圖形桌面字體變得異常大,這是正常情況,因為電腦沒有顯卡驅動,當重裝驅動后就正常了。
(三)cuda安裝
1、安裝命令
sudo apt-get install remove --purge nvidia*
(這條命令可以驅動和cuda一起安裝)
2、設置環境變量
終端中輸入:
sudo gedit /etc/profile
在打開的文件末尾,復制添加以下兩行:
export PATH=/usr/local/cuda-10.0/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
3、驗證是否安裝成功
驗證CUDA Toolkit,輸入:
nvcc -V
(四)cuda卸載
cuda的默認安裝在 /usr/local/cuda-8.0下,用下面的命令卸載:
sudo /usr/local/cuda-8.0/bin/uninstall_cuda-8.0.pl
會出現長時間的刷屏現象,說明cuda文件其實還蠻多的。
/usr/local/cuda-8.0沒有自動刪除,需要手動完成刪除。
下載安裝文件
按需求下載cudnn的安裝文件:https://developer.nvidia.com/rdp/cudnn-archive
(五)安裝cudnn
cp cudnn-10.0-linux-x64-v7.3.1.20.solitairetheme8 cudnn-10.0-linux-x64-v7.3.1.20.tgz tar -xvf cudnn-10.0-linux-x64-v7.3.1.20.tgz
解壓下載的文件,可以看到cuda文件夾,在當前目錄打開終端,執行如下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn.h sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
(3)查看cudnn版本
在終端輸入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2