Ubuntu16.04下安裝CUDA9.0和CUDNN7.0填坑

weixin_33724059發表於2019-01-08
UbuntuDNN
  1. 顯示出NVIDIA GPU版本資訊
lspci | grep -i nvidia
  1. 驗證自己的Linux版本是否支援 CUDA
uname -m && cat /etc/*release
  1. 驗證系統是否安裝了gcc，安裝則現實gcc版本
gcc –version

安裝過程遇到了好多坑，網上安裝教程很多，這裡就不詳述了，記錄一下檢測是否安裝成功的方法

安裝成功後的Test

  1. 檢視CUDA版本資訊
nvcc --version
或者
cat /usr/local/cuda/version.txt
3175603-f629d8bd334aab00.png
3175603-ea6da0ebadf5ccbd.png

在終端輸入下例命令

source /etc/bash.bashrc

echo $CUDA_HOME
3175603-067313d1c4fa6e3d.png
這樣就代表配置環境成功
  1. 檢視CUDNN版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

相關文章