我想在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
运行作业时出现的日志答案 0 :(得分:1)
免责声明:截至2017年11月1日,使用1.0以外的任何内容均未获得正式支持。
您需要指定启用GPU的TensorFlow版本:
REQUIRED_PACKAGES = ['tensorflow-gpu==1.3.0']
但pip
的版本已过时,因此您需要强制首先更新。