CAFFE（FAQ.1）：Ubuntu 配置caffe 框架之数据库读取错误：ImportError: No module named lmdb解决办法

Z:

在安装了caffe框架后需要读取大量的数据进行学习训练。比如在MNIST识别训练中，一般直接读图片会比较耗时，我们一般将图片转存为数据库中。目前主流的数据库有以下两种选择：

• 　　LevelDB
• 　　LmDB

我们可以通过如下代码测试读取LmDB数据库所用时间，具体代码如下

# coding: utf-8
import caffe
import numpy as np
import lmdb
import sys
from caffe.proto import caffe_pb2
def lmdb_process(db_path):
    env = lmdb.open(db_path)
    datum = caffe_pb2.Datum()
    item_id = 0
    with env.begin() as txn:
        cursor = txn.cursor()
        for key, value in cursor:
            datum.ParseFromString(value)
            label = datum.label
            img = caffe.io.datum_to_array(datum)
            # do something here
            item_id +=1
    print item_id
    
if __name__ == '__main__':
    db_path = sys.argv[1]
    lmdb_process(db_path)

如果运行没有出错，则在终端上执行如下代码，即可实现读取lmdb数据库所用时间。

cd /x...x/pythoncode/  #进入你存放read_lmdb.py的文件夹路径，然后执行

time python2 read_lmdb.py /home/pertor/caffe/examples/mnist/mnist_train_lmdb/

有时在运行以上读取数据库LMDB代码时会出现错误的情况，一般LMDB数据库读取错误提示：ImportError: No module named lmdb

此时在你电脑上可能存在两种情况：

（1） 电脑中只存在python2或者python3一种	sudo pip install lmdb
（2）电脑中同时存在python2和python3	注意在不同的版本中略有改变 python2 -m pip install 模块名称 python3 -m pip install 模块名称

当你电脑中同时存在python2和python3时，当caffe用的python2时（推荐），python2 -m pip install lmdb即可。如果用两种版本则python2 -m pip install lmdb、python3 -m pip install lmdb同时运行即可。

 具体如何知道电脑中存在那个python版本和自己电脑默认使用的Python版本，执行如下代码：

$ python2 --version
Python 2.7.14 :: Anaconda, Inc.
$ python --version #查看默认版本
Python 2.7.14 :: Anaconda, Inc.
$ python3 --version
Python 3.6.4 :: Anaconda, Inc.

总结：

　　作者的系统默认使用的python2，执行import lmdb出错提示：ImportError: No module named lmdb

 　　解决办法：python2 -m pip install lmdb（最好不要直接pip install lmdb）