ubuntu16.04 安裝cuda 10.1 和cudnn

1.nvidia-smi看看對應驅動的版本，如果沒有還得先去下載驅動，https://www.nvidia.com/，或者先查詢好顯卡型號和驅動、cuda版本對應的關系，然后進入第二步，下載對應的安裝包，包里有對應的驅動。

2.安裝cuda, https://developer.nvidia.com/cuda-downloads

 

sudo sh cuda_10.1.105_418.39_linux.run 安裝

如果已經安裝的驅動，那么把第一項“x”去掉，不然安不上。

設置環境變量：

sudo gedit ~/.bashrc

末尾輸入：

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

刷新環境變量：

source ~/.bashrc

 

查看cuda版本命令：cat /usr/local/cuda/version.txt

3.安裝cudnn，https://developer.nvidia.com/rdp/cudnn-download

 

 

如果找不到cuDNN Library for Linux，就點擊最下面的Archived cuDNN Releases，然后在里面找

 

cp cudnn-10.1-linux-x64-v7.6.2.24.solitairetheme8 cudnn-10.1-linux-x64-v7.6.2.24.tgz

解壓：

tar -zxvf cudnn-10.1-linux-x64-v7.6.2.24.tgz

拷貝：

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

 

查看版本：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2