安裝環境
- OS:Ubuntu 18.04 64 bit
- 顯卡:NVidia GTX 1080
任務:安裝 CUDA 10及cuDNN 7
工具下載
NVidia官網下載下列文件:
CUDA 10:cuda_10.0.130_410.48_linux.run
cnDNN 7.4:cudnn-10.0-linux-x64-v7.4.2.24.tgz
安裝CUDA
$ sudo sh cuda_10.0.130_410.48_linux.run
先輸入accept
接受協議,然后按需回答問題即可。
注意:
- 如果當前顯卡驅動版本高於CUDA安裝包內的驅動,建議跳過這一步,保留原來的顯卡驅動即可
- 為了節省空間,sample可以不裝
接着將下面內容追加到~/.bashrc
:
export PATH=/usr/local/cuda-10.0/bin:$PATH
驗證CUDA
使用nvidia-smi
查看驅動版本為415.23
:
$ nvidia-smi
Thu Jan 24 18:00:52 2019
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 415.23 Driver Version: 415.23 CUDA Version: 10.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 1080 Off | 00000000:01:00.0 On | N/A |
| 48% 35C P8 8W / 180W | 200MiB / 8116MiB | 2% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 1462 G /usr/lib/xorg/Xorg 83MiB |
| 0 1600 G /usr/bin/gnome-shell 100MiB |
| 0 1794 G /opt/teamviewer/tv_bin/TeamViewer 14MiB |
+-----------------------------------------------------------------------------+
CUDA版本為10.0
:
$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
安裝cuDNN
流程是:解壓,拷貝,配置環境變量
# 解壓
$ tar -zxvf cudnn-10.0-linux-x64-v7.4.2.24.tgz
# 拷貝
$ cd cudnn-10.0-linux-x64-v7.4.2.24
$ sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64
# 修改權限
$ sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*
將下面內容追加到~/.bashrc
:
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
export CUDNN_PATH="/usr/local/cuda-10.0/lib64/libcudnn.so"
驗證cnDNN
下面命令不報錯就OK。
$ echo -e '#include"cudnn.h"\n void main(){}' | nvcc -x c - -o /dev/null -lcudnn