1. 程式人生 > >Ubuntu16.04+cuda9.0安裝教程

Ubuntu16.04+cuda9.0安裝教程

Ubuntu16.04+cuda9.0安裝教程

轉載自:https://www.cnblogs.com/iloveblog/p/7683349.html

1.安裝NVIDIA驅動

首先去官網(http://www.nvidia.cn/Download/index.aspx?lang=cn)查詢適配自己電腦GPU的驅動,
  在這裡插入圖片描述

安裝驅動見上一篇部落格。裝好後輸入nvidia-smi,輸出gpu資訊證明安裝成功。

2.安裝CUDA9.0

首先去官網 https://developer.nvidia.com/cuda-toolkit-archive
  下載CUDA 9.0,下載的是.run檔案,下載for Linux 版本。

下載完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 -j4(這是多執行緒,可以根據自己電腦配置選,也可以不要-j4,那就是單執行緒)

./deviceQuery

如果顯示的是關於GPU的資訊,則說明安裝成功了。