[筆記] Ubuntu 18.04安裝cuda 10及cudnn 7流程


安裝環境

  • 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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM