caffe升级到cuda10.2 + cudnn 8方法

原创已于 2023-02-16 10:29:17 修改 · 588 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#caffe #深度学习 #人工智能

于 2022-05-14 21:06:48 首次发布

AIoT 同时被 2 个专栏收录

19 篇文章 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

Jetson

16 篇文章

订阅专栏

在将Caffe环境更新到CUDA 10.2和cuDNN 8的过程中，需要对include/caffe/util/device_alternate.hpp中的头文件顺序、src/caffe/layers/reduce.cu中的__shfl_up相关操作以及src/caffe/layers/cudnn_conv_layer.cpp和DenseBlock_layer.cu中的cudnn算法选择函数进行修改。这些改动对于成功编译和运行使用新版本CUDA和cuDNN的Caffe至关重要。

将caffe的环境升级到cuda10.2 + cudnn8时，需进行下列修改，否则无法完成编译。

1）include/caffe/util/device_alternate.hpp

将头文件#include <driver_types.h>放置在前面，如下：

#include <driver_types.h>  // cuda driver types

#include <cublas_v2.h>

#include <cuda.h>

#include <cuda_runtime.h>

#include <curand.h>

#ifdef USE_CUDNN  // cuDNN acceleration library.

#include "caffe/util/cudnn.hpp"

2）__shfl_up和__shfl_up修改

1> src/caffe/layers/reduce.cu

Line 44 to : shuff = __shfl_down_sync(0xFFFFFFFF, x, offset);

2> include/contrib/moderngpu/include/device/intrinsics.cuh

Line 115 to :

了解本专栏