1 查询cuDNN 安装情况
cuDNN是GPU加速计算深层神经网络的库。
突然发现cudnn没安装上:
$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
得到:
/usr/local/cuda/include/cudnn.h
为了查证是否是自己记错目录,查询一下
$ find -name cudnn.h
发现确实没有安装好
2 查询CUDA版本
$ nvcc --version
Cuda compilation tools, release 9.0, V9.0.176
CUDA 版本是9.0
附:查询 tensorflow , Keras 版本 (可能用得到)
$ pip3 freeze | grep tensorflow
tensorflow==1.8.0
tensorflow-gpu==1.8.0
$ pip3 freeze | grep Keras
Keras==2.2.0
Keras-Applications==1.0.2
Keras-Preprocessing==1.0.1
3 官网下载并安装
去官网
下载cuDNN&#x