caffe 编译错误:too few argument in function call
error log:
caffe/include/caffe/util/cudnn.hpp(112): error: too few arguments in function call
cudnn版本问题,解决:
添加判断,不同版本参数不同
#if CUDNN_VERSION_MIN(6, 0, 0)
CUDNN_CHECK(cudnnSetConvolution2dDescr...
原创
2019-05-05 16:37:16 ·
2189 阅读 ·
0 评论