CUDA/cuDNN（ubuntu18.04 + 1050Ti）

發現有些指令直接複製運行有問題，可能需要重新在終端敲一下

1、顯卡版本查看：

ubuntu-drivers devices


可以看到推薦的驅動版本（我的是390）

2、禁用 nouveau

sudo gedit /etc/modprobe.d/blacklist.conf


最後添加

blacklist nouveau
options nouveau modeset=0


保存退出

sudo update-initramfs -u


3、安裝Nvidia GPU驅動

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia‐390 #這裡要寫自己適合的gpu驅動版本


以下兩步顯示已經安裝過：

sudo apt install mesa‐common‐dev
sudo apt install freeglut3‐dev
sudo apt upgrade #時間較長
sudo reboot #重啟


驗證：

sudo nvidia-smi
sudo nvidia-settings


關閉更新：

sudo apt-mark hold nvidia-390


4、gcc降級

sudo apt-get install gcc-4.8
sudo apt-get install g++-4.8
cd /usr/bin
sudo rm gcc
sudo ln -s gcc-4.8 gcc
sudo rm g++
sudo ln -s g++-4.8 g++


查看版本，如提示版本為4.8則成功

gcc --version
g++ --version


PS：g++多版本管理

查看已裝版本（我這裡有4.87）

ls /usr/bin/gcc-*


把兩個版本添加如下：（最後一項是優先順序）

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-7 100
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 50
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.8 50


切換版本：

sudo update-alternatives --config gcc
sudo update-alternatives --config g++


5、CUDA安裝

去官網CUDA Toolkit Archive下載相應版本，選擇9.0，支持16.04

安裝包移動到home

sudo sh cuda_9.0.176_384.81_linux.run


一直按回車

提問：

You are attempting to install on an unsupported configuration 選yes

安裝時需要注意，有一個提示是否安裝顯卡驅動，因為之前已經安裝顯卡驅動，所以選擇no

其它選項都默認

安裝成功後，配置環境

sudo gedit ~/.zshrc # 如果用bash: ~/.bashrc


在文件的最後添加

export PATH=/usr/local/cuda-9.0/bin${PATH:+:$PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}


其中cuda要修改為對應的版本號

驗證：（我的先重啟生效的）

cd NVIDIA_CUDA-9.0_Samples/5_Simulations/fluidsGL
make clean && make
./fluidsGL


正常運行則安裝成功：

6、cuDNN安裝

需要先註冊NVIDIA賬號

官網下載合適版本

安裝可以參考Installation Guide：

Install the runtime library, for example:

sudo dpkg -i libcudnn7_7.0.3.11-1+cuda9.0_amd64.deb


Install the developer library, for example:

sudo dpkg -i libcudnn7-dev_7.0.3.11-1+cuda9.0_amd64.deb


Install the code samples and the cuDNN Library User Guide, for example:

sudo dpkg -i libcudnn7-doc_7.0.3.11-1+cuda9.0_amd64.deb


驗證：

將/usr/src/cudnn_samples_v7複製到可寫目錄：

cp -r /usr/src/cudnn_samples_v7 $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN


顯示Test passed 則安裝成功：