ubuntu16 + gt940m + cuda9.2 + cudann7.3

1、裝顯卡驅動

檢測顯卡型號和對應推薦安裝驅動

root$ ubuntu-drivers devices

自動安裝所推薦的驅動

root$ sudo ubuntu-drivers autoinstall

 

確保BIOS中Secure Boot設置爲disabled，
重啓檢查驅動是否安裝成功
root$ nvidia-smi

2、裝CUDA
鏈接：https://developer.nvidia.com/cuda-toolkit-archive
CUDA推薦下載.run可以根據提示安裝

執行如下命令：

sudo sh cuda_9.0.176_384.81_linux.run

然後用一本書的書角壓住回車鍵，直到服務條款顯示到100%。接着按下面的步驟選擇：

accept

n（不要安裝driver）

y

y

y

安裝完成後，設置環境變量。

      （3）設置環境變量。

打開主目錄下的 .bashrc文件添加如下路徑，例如我的.bashrc文件在/home/wangyuanwei下，如果沒有找到，則按Ctrl+H鍵顯示隱藏文件。

    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
    export PATH=$PATH:/usr/local/cuda-9.0/bin
    export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

終端運行：source ~/.bashrc

檢查：nvcc --version
 

3.cudnn的安裝

cudnn的安裝非常簡單，

（1）下載安裝文件

按需求下載cudnn的安裝文件：https://developer.nvidia.com/rdp/cudnn-archive

（2）安裝cudnn

解壓下載的文件，可以看到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