Cloud ML Tensorflow和Cudnn版本的兼容性

时间:2017-11-01 23:44:47

标签: tensorflow tensorflow-gpu google-cloud-ml cudnn

我想在Cloud ML上使用Tensorflow 1.3(可能是1.4)。我在Cloud ML上的多GPU机器上运行作业

我是通过在setup.py中指定tensorflow版本来实现的,如下所示:

from setuptools import setup

REQUIRED_PACKAGES = ['tensorflow==1.3.0']

setup(
    name='my-image-classification',
    install_requires=REQUIRED_PACKAGES,
    version='1.0',
    packages=['my_image_classification',
              'my_image_classification/foo',
              'my_image_classification/bar',
              'my_image_classification/utils'],
    )

Cloud ML上安装的cudnn库是什么?它与tensorflow 1.3和tensorflow 1.3+兼容吗?

我能够开始工作,但性能比预期值低10倍,如果图书馆的基础链接出现问题,我很好奇

编辑:

我很自信现在Cloud ML上的Cudnn版本与Tensorflow 1.3所需的版本不匹配。我注意到Tensorflow 1.3工作缺少"创建Tensorflow设备(/ gpu:0 ...)"使用默认的可用Tensorflow on cloud ml

运行作业时出现的日志

1 个答案:

答案 0 :(得分:1)

免责声明:截至2017年11月1日,使用1.0以外的任何内容均未获得正式支持。

您需要指定启用GPU的TensorFlow版本:

REQUIRED_PACKAGES = ['tensorflow-gpu==1.3.0']

pip的版本已过时,因此您需要强制首先更新。