Google Colab GPU加速适用于2.x，但不适用于1.x
Google Colab GPU加速适用于2.x，但不适用于1.x
提问于 2019-12-25 13:53:08
在https://colab.research.google.com/notebooks/gpu.ipynb中，我假设这是谷歌的图形处理器加速的官方演示，如果我遵循这些步骤，使用Tensorflow 2.x的图形处理器加速(大约比CPU快60倍)工作。但是，如果我想使用1.15版本，就像在https://colab.research.google.com/drive/12dduH7y0GPztxSM0AFlfpjj8FU5x8YSv中一样(与第一个链接中的notebook相比，唯一的变化是两次都去掉了"%tensorflow_version 2.x“)，tf.test.gpu_device_name()返回字符串/device:GPU:0，但没有加速。我真的很喜欢使用1.5和1.15之间的Tensorflow版本，因为我想运行的代码使用了Tensorflow 2.x中删除的函数。有谁知道如何在使用Tensorflow 1.x的同时获得GPU加速？
回答 1
Stack Overflow用户
回答已采纳
发布于 2019-12-25 14:39:31
在你的notebook中，你的代码实际上并没有被执行，因为你没有调用session.run()或者tf.enable_eager_execution()。
在您的代码顶部添加tf.enable_eager_execution()，您将看到cpu和gpu时间的真正区别。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/59475593
复制相关文章
相似问题
腾讯云开发者
