1.安裝NVIDIA驅動
首先去官網(http://www.nvidia.cn/Download/index.aspx?lang=cn)查找適配自己電腦GPU的驅動,我的電腦驅動版本如下:
執行如下語句安裝NVIDIA驅動
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384 //驅動版本不同,對應不同的數字
sudo apt-get install mesa-commoon-dev //本句及下一句命令有的電腦需要,有的電腦不需要,根據實際情況輸入命令
sudo pat-get install freeglut3-dev
執行完上述命令后若無問題,重啟,若有問題,單獨查資料解決
sudo reboot
重啟后輸入:
sudo nvidia-smi
如果顯示GPU列表,則證明驅動安裝成功了,另外也可以通過如下命令:
sudo nvidia-settings
查看GPU信息,顯示界面如下:
2.安裝CUDA9.0
首先去官網下載CUDA 9.0,這里因為寫該博客時NVIDIA網站進不去了,就截了一個圖,下載的是.run文件
下載完CUDA 9.0之后執行如下語句,運行.run文件
sudo sh cuda_9.0.176_384.81_linux.run
單擊回車,一路往下運行,直到提示“是否為NVIDIA安裝驅動nvidia-384?”,選擇否,因為已經安裝好驅動程序了,其他的全都是默認,不過要記住安裝位置,默認是安裝在/usr/local/cuda文件夾下。
配置環境變量,運行如下命令打開profile文件
sudo gedit /etc/profile
打開文件后在文件末尾添加路徑,也就是安裝目錄,命令如下:
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
保存,然后重啟電腦
sudo reboot
3.測試CUDA的Samples例子
cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
如果顯示的是關於GPU的信息,則說明安裝成功了。