Использование tensorflow.contrib.data.Dataset в Cloud ML

Недавно я изменил свой конвейер данных в тензорном потоке с потоковой обработки на новый API набора данных, что довольно удобно, если вы хотите проверять свою модель каждую эпоху.

Я заметил, что текущая версия tensorflow во время выполнения в Cloud ML 1.2. Тем не менее, я пытался использовать ночную сборку tensorflow v1.3, но установка pip завершается с ошибкой:

AssertionError: tensorflow==1.3.0 .dist-info directory not found
Command '['pip', 'install', '--user', '--upgrade', '--force-reinstall', '--no-deps', u'tensorflow-1.3.0-cp27-none-linux_x86_64.whl']' returned non-zero exit status 2

Кому-нибудь удалось использовать tensorflow.cotrib.data.Dataset с движком Cloud ML?


google-cloud-platform tensorflow google-cloud-ml tensorflow-gpu google-cloud-ml-engine
person Vlad Shkola    schedule 21.09.2017    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Это сработало для меня: создайте файл setup.py со следующим содержимым:

from setuptools import find_packages
from setuptools import setup

REQUIRED_PACKAGES = ['tensorflow==1.3.0']

setup(
    name='trainer',
    version='0.1',
    install_requires=REQUIRED_PACKAGES,
    packages=find_packages(),
    include_package_data=True,
    description='Upgrading tf to 1.3')

дополнительные сведения о файле setup.py доступны по адресу: Упаковка учебного приложения

person Jan Krynauw    schedule 27.09.2017