1. CUDA和CUDNN
cuda下载地址:https://developer.nvidia.com/cuda-toolkit-archive
cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-archive
注意当前版本caffe只支持 cuda8.0-10.2 和 cudnn v7 版本,对照自己的cuda版本下载对应的cudnn
cuda的安装过程网上都有我就不详述了,讲一下cudnn如何安装,下载好cudnn后,解压出来:
tar -xzvf cudnn-10.2-linux-x64-v7.6.5.32.tgz
如何把cudnn的文件移到cuda中:
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
显示CUDNN_MAJOR 7则说明cudnn安装成功:
#define CUDNN_MAJOR 7 #define CUDNN_MINOR 6 #define CUDNN_PATCHLEVEL 5 -- #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL) #include "driver_types.h"
2. caffe安装
当前caffe版本(https://github.com/BVLC/caffe):