如果我們要進(jìn)行深度學(xué)習(xí)倦始，CUDA 是一個(gè)必不可少的工具榆综，那么在 Ubuntu 系統(tǒng)中應(yīng)該如何安裝 CUDA 呢您觉？
安裝好 Ubuntu 系統(tǒng)之后，默認(rèn)是沒有安裝顯卡驅(qū)動(dòng)的拖吼，因此我們需要手動(dòng)安裝顯卡驅(qū)動(dòng)
1. 查看自己的顯卡情況鳞上，確定自己的顯卡是可以被檢測(cè)到的
在命令行輸入 lspci，即可看到自己的顯卡情況
可以看到我的顯卡是 GTX 1050 的吊档，由于沒有安裝顯卡驅(qū)動(dòng)篙议，輸入 nvidia-smi 命令是無(wú)效的
此時(shí)可以安裝顯卡驅(qū)動(dòng)，打開 “軟件和更新”怠硼，如下圖所示
注意：在安裝之前鬼贱，一定要先禁用 security boot 模式，不然可能會(huì)報(bào)錯(cuò)香璃！
安裝驅(qū)動(dòng)之后侵歇，要先重啟電腦 reboot笆檀，然后打開一個(gè)新的終端，鍵入命令：nvidia-smi酗洒，如果出現(xiàn)如下所示的信息枷遂，則表示安裝成功酒唉！
可以看到驅(qū)動(dòng)的版本號(hào)為 440.95.01，對(duì)應(yīng)的 CUDA 版本為 10.2侄榴，當(dāng)然 10.2 以下的也是可以安裝的
2. 下載 CUDA 和 cudnn (深度學(xué)習(xí)的加速庫(kù))
首先网沾，我們得從 NVIDIA 的官網(wǎng)來(lái)下載 CUDA辉哥，打開百度搜索 CUDA 10.1，這里根據(jù)個(gè)人的需要選擇 CUDA 的版本
選擇 Ubuntu 版本的 64 位的 runfile(local) 文件即可
然后下載 cudnn，百度搜索 cudnn 即可去官網(wǎng)下載
點(diǎn)擊第二個(gè)進(jìn)入官網(wǎng)钉凌，選擇版本為 7.5.1 的捂人，這里根據(jù)自己的需要來(lái)選擇對(duì)應(yīng)的版本
3. 安裝 CUDA
此外席爽，在安裝 CUDA 之前需要禁用 security boot只锻，如何禁用可以百度教程，不然可能會(huì)導(dǎo)致驅(qū)動(dòng)安裝失敗等問(wèn)題捐寥！
另外需要做的一件事情就是禁用 nouveau 的驅(qū)動(dòng)祖驱，首先使用命令：lsmod | grep nouveau 查看 nouveau 是否被禁用捺僻，如果什么顯示都沒有，則說(shuō)明被禁用就珠。如果 nouveau 并沒有被禁用醒颖，可以參考這篇文章：https://blog.csdn.net/yinwangde/article/details/89439648
首先執(zhí)行官網(wǎng)提供的命令
sudo sh cuda_10.1.105_418.39_linux.run
4. 配置環(huán)境變量
一般系統(tǒng)安裝的軟件默認(rèn)存儲(chǔ)在 /usr/local 目錄下拆檬，我們也可以使用命令來(lái)進(jìn)行查看
查看的命令一般為：which cuda 或者 whereis cuda竟贯，如果不確定可以都試一下
我們進(jìn)入到 cuda-10.1 的 bin 目錄下，并復(fù)制該路徑晌缘，進(jìn)行環(huán)境變量的配置
在命令行輸入 sudo gedit ~/.bashrc磷箕，在打開的文件末尾輸入 export PATH=/usr/local/cuda-10.1/bin:$PATH
保存之后，更新環(huán)境變量 source ~/.bashrc
然后測(cè)試配置環(huán)境變量是否成功芒填，輸入命令 nvcc -V
如上圖所示殿衰，說(shuō)明環(huán)境變量配置成功盛泡！
5. 配置 cudnn
首先解壓該文件，解壓的命令為 tar -zxvf cudnn-10.1-linux-x64-v7.5.1.10.tgz
將 cuda/include/cudnn.h 文件復(fù)制到 /usr/local/cuda/include 文件下
將 cuda/lib64/ 下所有文件復(fù)制到 /usr/local/cuda/lib64 文件夾
復(fù)制的命令為：sudo cp * /usr/local/cuda/lib64 （* 表示所有的文件）
最后一步箱硕，為上述文件讀取和可執(zhí)行權(quán)限
sudo chmod 755 /usr/local/cuda/include/cudnn.h
sudo chmod 755 /usr/local/cuda/lib64/libcudnn*
6. 為了保險(xiǎn)起見，我們需要測(cè)試以下 gpu 是否可用
我們先安裝 gpu 版本的 TensorFlow舰罚，版本可以選擇 2.0 以上的
首先打開終端薛耻，輸入 ipython
導(dǎo)入 TensorFlow 的庫(kù)
這里出現(xiàn)錯(cuò)誤饼齿，是因?yàn)?沒有安裝 TensorRT蝙搔，但是不影響 tensorflow 的使用
測(cè)試 gpu 是否可用：tf.test.is_gpu_available()吃型，如果出現(xiàn) true，則表示 gpu 可用
