Кафе не импортируется в питоне - PullRequest
1 голос
/ 12 марта 2019

Ubuntu 18.04

Python 2.7

Моя проблема в том, что я не могу импортировать модуль caffe в python, даже если я его установил.Я считаю, что это проблема переменной пути / env.

rivaldo4t@Rivaldo-OS3:~$ python
Python 2.7.15rc1 (default, Nov 12 2018, 14:31:15) 
[GCC 7.3.0] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import caffe
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: No module named caffe

Как указано в здесь , я запустил команду sudo apt install caffe-cpu для установки предварительно скомпилированного Caffe, и она прошла успешно.Когда я набираю caffe в терминале, я получаю что-то вроде

rivaldo4t@Rivaldo-OS3:~$ caffe
caffe: command line brew
usage: caffe <command> <args>

commands:
  train           train or finetune a model
  test            score a model
  device_query    show GPU diagnostic information
  time            benchmark model execution time

, что означает, что caffe установлено правильно.Как уже упоминалось здесь , я попытался установить в моей переменной PYTHONPATH расположение каталога установки Caffe.Итак, я сделал это,

rivaldo4t@Rivaldo-OS3:~$ which caffe
/usr/bin/caffe

Когда я попытался запустить echo $ PYTHONPATH, он оказался пустым (я до сих пор не знаю, почему).Затем я попытался обновить путь, используя

caffe_root = '/usr/bin/caffe/python'
import sys
sys.path.insert(0, caffe_root)

Так что теперь я получаю это при печати sys.path -

rivaldo4t@Rivaldo-OS3:~$ python -c 'import sys; print(sys.path)'
['', '/home/rivaldo4t', '/usr/bin/caffe/python', '/usr/lib/python2.7', '/usr/lib/python2.7/plat-x86_64-linux-gnu', '/usr/lib/python2.7/lib-tk', '/usr/lib/python2.7/lib-old', '/usr/lib/python2.7/lib-dynload', '/home/rivaldo4t/.local/lib/python2.7/site-packages', '/usr/local/lib/python2.7/dist-packages', '/usr/lib/python2.7/dist-packages']

Но я все еще не могу импортировать caffe в python.Еще я заметил, что / usr / bin / caffe не является каталогом, и я не могу понять, где установлен caffe, если не здесь.

Любая помощь приветствуется.Спасибо

1 Ответ

1 голос
/ 12 марта 2019

Когда вы устанавливаете caffe в Ubuntu, используя sudo apt install caffe-cpu, он компилирует привязки только для Python 3 (_caffe.cpython-36m-x86_64-linux-gnu.so), который расположен в /usr/lib/python3/dist-packages/caffe/.Поэтому короткий ответ - использовать вместо этого python 3.

Длинный ответ - скомпилировать caffe с привязками python 2 из источника.

...