我正在通过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'。
请帮忙.. 提前谢谢。
答案 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.'
)