我的caffe 一直没有使用cudnn 加速
怎么使用cudnn 加速:
A. 首先知道自己cuda的版本是多少
nvcc -V 185 服务器
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Sun_Sep__4_22:14:01_CDT_2016
Cuda compilation tools, release 8.0, V8.0.44
可以看到cuda 是8.0 的版本
泰坦 black 的cuda 版本为:
cuda 7.5 的版本
在网上下载相应的版本按照说明就可以搞定了。