查看cuda cudnn版本
查看cuda版本:
cat /usr/local/cuda/version.txt
或者
nvcc -V
查看cudnn版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
1. 安装cuda9.0和cudnn7
参考:https://www.codetd.com/article/727311
安装cuda9.0
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
安装cudnn7
tar -xzvf cudnn-9.0-linux-x64-v7.1.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 /usr/local/cuda/lib64/libcudnn*
2. 安装cudnn7并切换软链接
安装了cuda-8.0和cudnn-v6,cuda-9.0,需要安装cudnn-v7,并修改cuda软链接。
步骤:
解压cudnn-v7后将其复制到对应版本的cuda-v9.0中,并赋予权限:
xxq@linux:~$ sudo cp /home/xxq/cudnn/cuda/include/* /usr/local/cuda-9.0/include/
xxq@linux:~$ sudo cp /home/xxq/cudnn/cuda/lib64/* /usr/local/cuda-9.0/lib64/
xxq@linux:~$ sudo chmod +x /usr/local/cuda-9.0/include/cudnn.h
xxq@linux:~$ sudo chmod +x /usr/local/cuda-9.0/lib64/libcudnn*
将cuda软链接指向cuda-9.0,查看cuda的版本为9.0,查看cudnn的版本为7.0.3,即修改成功
xxq@linux:~$ cd /usr/local
xxq@linux:/usr/local$ sudo rm -rf cuda
xxq@linux:/usr/local$ sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
xxq@linux:/usr/local$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176
xxq@linux:/usr/local$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 3
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
xxq@linux:/usr/local$