Question

最近我将tensorflow中的数据管道从线程更改为新的Dataset api，一旦你想在每个时代验证你的模型，这非常方便。

我注意到current runtime version of tensorflow in Cloud ML is 1.2。尽管如此，我还是尝试使用nightly build of 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引擎一起使用？

Answer 1

这对我有用：创建一个包含以下内容的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文件的更多信息，请访问：Packaging a Training Application

在Cloud ML

1 个答案: