Linux(Ubuntu20.04)安装配置Caffe环境(No module named caffe)
2023-06-13 09:12:22 时间
0. 前言
ImportError: No module named caffe
我滴天,这咖啡真不好喝!
我感觉我为了解决这个问题,都能写一篇论文了,整整搞了两天,是真麻烦,所以,我建议,如果能换成不用caffe的代码的话,尽量别用caffe跑了,太难了~
1. 尝试解决办法
首先,网上推荐的是用编译的方式进行安装配置caffe。具体如下:
1.1 创建环境
conda create -n caffe_env python=2.7.18
conda activate caffe_env
pip install pip -U
pip install numpy
1.2 安装依赖
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev
sudo apt install libatlas-base-dev
sudo apt install libgflags-dev libgoogle-glog-dev liblmdb-dev
1.3 配置caffe
下载 caffe
解压 unzip caffe-master.zip
进入caffe根目录:
复制Makefile.config.bak为Makefile.config
修改Makefile.config和Makefile文件。
Makefile.confg参考如下:
USE_CUDNN := 1
OPENCV_VERSION := 3
CUDA_DIR := /usr/local/cuda
# 如果cuda版本大于等于9.0,则注释以下两行:
# -gencode arch=compute_20,code=sm_20 \
# -gencode arch=compute_20,code=sm_21
CUDA_ARCH := -gencode arch=compute_30,code=sm_30 \
-gencode arch=compute_35,code=sm_35 \
-gencode arch=compute_50,code=sm_50 \
-gencode arch=compute_52,code=sm_52 \
-gencode arch=compute_60,code=sm_60 \
-gencode arch=compute_61,code=sm_61 \
-gencode arch=compute_61,code=compute_61
BLAS := atlas
ANACONDA_HOME := $(HOME)/conda/envs/py3_caffe_source
PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
$(ANACONDA_HOME)/include/python3.6m \
$(ANACONDA_HOME)/lib/python3.6/site-packages/numpy/core/include
PYTHON_LIB := $(ANACONDA_HOME)/lib
WITH_PYTHON_LAYER := 1
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
BUILD_DIR := build
DISTRIBUTE_DIR := distribute
TEST_GPUID := 0
Q ?= @
Makefile内容主要修改以下两项:
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
PYTHON_LIBRARIES ?= boost_python3 python3.6m
1.4 编译caffe
make all -j8
make test -j8
make pycaffe -j8
# 如果报错 需重新编译 make clean
1.5 收尾
拷贝编译后的libcaffe.so至caffe的虚拟环境 在caffe源码项目下/python目录下:
cp -rf caffe/ ~/miniconda3/envs/py3_caffe_source/lib/python3.6/
cp -rf ../.build_release/lib/* ~/miniconda3/envs/py3_caffe_source/lib/
进入conda虚拟环境安装caffe依赖的Python库
pip3 install numpy
pip3 install scikit-image
pip3 install protobuf
大概流程就这样,具体报错就网上搜吧。
2 大杀器!完美解决,简单方便
2.1 安装环境
conda create -n caffe_env python=3.6.10
conda activate caffe_env
conda install -c defaults caffe-gpu
2.2 测试环境
(caffe_env) user@user-Ubuntu:~/caffe_env$ python
Python 3.6.10 |Anaconda, Inc.| (default, May 5 2021, 11:02:1)
[GCC 8.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import caffe
>>> exit()
如果不报错,就成了,我完成时成了,挣了两天的活,最后,一句命令搞定了,既开心又生气。 反正希望能帮到你们,能解决问题最好,每个人的情况不一样,我建议首先试试第2步的大杀器,太棒了。
相关文章
- 处理Linux下子进程结束处理方法研究(linux子进程结束)
- Linux中的i节点:储存文件信息的构建块(linux中i节点)
- Linux远程桌面连接:轻松实现跨距离协作(linux连接远程桌面)
- 安装及使用Linux系统中SSH安装及使用指南(linux中ssh)
- 安装 Linux下的G 编译器(linux安装g编译器)
- 从零开始:掌握 Linux 泛解析技能(泛解析linux)
- 命令探索Linux中head命令的奥秘(在linux中head)
- Linux赋权限的正确方法(linux如何赋权限)
- Linux系统安装:从刻录系统盘开始(刻录linux系统盘)
- Linux的千变万化:分支之旅(linux的分支)
- Linux安装LZMA:实现快速、稳健的文件压缩(linux安装lzma)
- Linux U盘启动:轻松实现操作系统安装(linux的u盘启动)
- 实现自由:Linux移动硬盘安装指南(把linux装到移动硬盘)
- Linux系统下NS2的安装与运行(linux安装ns2)
- 探秘Linux硬解码技术: 提升视频播放性能的捷径(linux硬解码)
- Linux如何实现RAID5?一起来了解RAID5的配置及使用方法(linux做raid5)
- Linux如何撤销上一步操作(linux撤销上一步操作)
- 如何在AMD处理器上安装Linux系统?(amd安装linux)
- Linux系统使用repo工具来安装软件包(linux安装repo)
- Linux跨平台移植实践指南(linux如何移植)
- Linux进程:探究运行状态(linux进程运行状态)
- 500错误:Linux服务器出现问题(linux错误500)
- Linux如何修改文件执行权限(linux修改执行权限)
- Linux开放指定端口:一次性完成任务(linux 开放指定端口)
- Linux 下简单步骤安装MySQL(linux 装mysql)
- 自制Linux系统:解锁你的开源世界(自己制作 linux系统)
- Mac OS和Linux系统的分别之处(mac和linux的区别)