Caffe安装问题汇总

最新推荐文章于 2022-06-17 22:09:35 发布

Doral

最新推荐文章于 2022-06-17 22:09:35 发布

阅读量2.4w

点赞数 2

分类专栏： caffe

本文链接：https://blog.csdn.net/doral/article/details/45620857

版权

本文汇总了Caffe安装过程中常见的问题及解决方案，包括CUDA安装、cuDNN编译错误、OpenBLAS找不到、g++版本过高以及pycaffe的可视化问题。通过提供的解决方法，可以有效地解决这些问题。

摘要由CSDN通过智能技术生成

ImportError: No module named skimage

【问题1】对于双显卡的笔记本而已，通过*.run方式安装CUDA时，重启后会黑屏出现无法进入桌面，只能进入tty

解决方法：采用*.deb方式来安装CUDA，不需要另外安装NVIDIA驱动（CUDA-7.0自带的就是最新的驱动），同时也不需关闭lightdm服务进入tty终端来安装。

在CUDA官网 https://developer.nvidia.com/cuda-downloads下载deb离线包安装即可。

【问题2】error: ‘CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING’ was not declared in this scope（安装cuDNN时，make 未声明错误）

In file included from ./include/caffe/util/device_alternate.hpp:40:0,
                 from ./include/caffe/common.hpp:19,
                 from ./include/caffe/util/math_functions.hpp:9,
                 from src/caffe/util/im2col.cpp:6:
./include/caffe/util/cudnn.hpp: In function ‘void caffe::cudnn::createPoolingDesc(cudnnPoolingStruct**, caffe::PoolingParameter_PoolMethod, cudnnPoolingMode_t*, int, int, int, int, int, int)’:
./include/caffe/util/cudnn.hpp:117:13: error: ‘CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING’ was not declared in this scope
     *mode = CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING;
             ^
Makefile:510: recipe for target '.build_release/src/caffe/util/im2col.o' failed
make: *** [.build_release/src/caffe/util/im2col.o] Error 1
make: *** Waiting for unfinished jobs....

解决方法：原因是之前用cuDNN版本cudnn-6.5-linux-x64-R2-rc1.tgz过低，不适合当前的caffe版本，在cuDNN官网https://developer.nvidia.com/cuDNN注册下载目前最新的cuDNN版本cudnn-6.5-linux-x64-v2.tgz即可。

【问题3】Check failed: status == CUDNN_STATUS_SUCCESS (6 vs. 0) CUDNN_STATUS_ARCH_MISMATCH （make runtest 的check错误）