导入错误matplotlib.pyplot

时间:2017-11-09 07:57:40

标签: google-cloud-platform google-cloud-ml

我正在通过Cloud-VM实例使用Cloud Machine Learning Engine开发对象检测器。遵循教程(https://cloud.google.com/blog/big-data/2017/06/training-an-object-detector-using-cloud-machine-learning-engine)。

当我提交以下培训工作时,我在Google Cloud Platform上收到了模块导入错误:

gcloud ml-engine jobs submit training `whoami`_object_detection_`date +%s` \
    --job-dir=${YOUR_GCS_BUCKET}/train \
    --packages dist/object_detection-0.1.tar.gz,slim/dist/slim-0.1.tar.gz \
    --module-name object_detection.train \
    --region us-central1 \
    --config object_detection/samples/cloud/cloud.yml \
    -- \
    --train_dir=${YOUR_GCS_BUCKET}/train \
    --pipeline_config_path=${YOUR_GCS_BUCKET}/data/faster_rcnn_resnet101_coco.config

错误如下:

...object_detection/utils/visualization_utils.py", line 24, in <module>
import matplotlib.pyplot as plt
ImportError: No module named matplotlib.pyplot

我已经使用pip install安装了matplotlib。 这段代码很好用python2.7 -c'import matplotlib.pyplot作为plt'。

请帮忙.. 提前谢谢。

1 个答案:

答案 0 :(得分:3)

通过在setup.py文件中添加包依赖项来解决此问题。

from setuptools import find_packages
from setuptools import setup

REQUIRED_PACKAGES = ['Pillow>=1.0','matplotlib', ]

setup(
    name='trainer',
    version='0.1',
    install_requires=REQUIRED_PACKAGES,
    packages=find_packages(),
    include_package_data=True,
    description='My trainer application package.'
)