PS: 簡單分享一下本人裝cuda的過程,個人覺得這個cuda安裝相比顯卡驅動安裝更加順利,沒那么多問題跑出,Cuda的版本就不糾結了哈,我直接用自己裝過的版本。
Cuda安裝( 參考:https://www.cnblogs.com/iloveblog/p/7683349.html)
1. 首先在cuda的官網將cuda的版本下載好,cuda的版本一般不選擇最新版,先看自己的顯卡裝的版本是怎樣的,參考下圖.

2. 下面是cuda9.0的安裝。
a. 去官網下載CUDA 9.0,下載的是.run文件

3.下載完CUDA 9.0之后執行如下語句,運行.run文件
sudo sh cuda_9.0.176_384.81_linux.run
4. 單擊回車,一路往下運行,直到提示“是否為NVIDIA安裝驅動nvidia-384?”,選擇否,因為已經安裝好驅動程序了,其他的全都是默認,不過要記住安裝位置,默認是安裝在/usr/local/cuda文件夾下。
5.配置環境變量,運行如下命令打開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
6.保存,然后重啟電腦
sudo reboot
7.測試CUDA的Samples
cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
如果顯示的是關於GPU的信息,則說明安裝成功了。
注意: 安裝cuda前,安裝顯卡驅動時,一定要將gcc make 等裝好。
Cuda卸載
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
sudo rm -rf /usr/local/cuda-8.0/
或者:
sudo apt-get remove cuda
sudo apt-get autoclean
sudo apt-get remove cuda*
Cudnn 安裝
1.下載得到cudnn-10.0-linux-x64-v7.4.2.24.tgz
2. 然后對它進行解壓,命令如下:
tar -zxvf cudnn-10.0-linux-x64-v7.4.2.24.tgz
3. 解壓之后可以得到以下文件:
cuda/include/cudnn.h
cuda/NVIDIA_SLA_cuDNN_Support.txt
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.7
cuda/lib64/libcudnn.so.7.4.2
cuda/lib64/libcudnn_static.a
4. 使用以下兩條命令復制這些文件到CUDA目錄下:
cp cuda/lib64/* /usr/local/cuda-10.0/lib64/
cp cuda/include/* /usr/local/cuda-10.0/include/
5.步驟4完成后,使用以下命令查看CUDNN的版本信息:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2