时间戳

首页 > 解决方案 > 版本 2+ 中 tensorflow-gpu 包的用途

tensorflow - 版本 2+ 中 tensorflow-gpu 包的用途

问题描述

虽然，我已经使用tensorflow了一段时间了，但最近当另一个 pip 模块弄乱了我的虚拟 tensorflow 环境时，我感到很困惑。我注意到现在我的tensorflow-gpu系统上安装了一个软件包（错误地认为它在tensorflow1.15 版本之后已被弃用）。

pip list | grep tensorflow

张量流估计器 2.5.0张量
流 GPU 2.5.0

但有些功能不起作用。例如使用：

import tensorflow.compat.v1 as tf

产生错误：

ModuleNotFoundError：没有名为“tensorflow.compat”的模块

所以，我不能使用兼容模式。

无论如何，我的问题是，由于官方 tensorflow 文档指出，homonym 包包含对 1.15 以上版本的 GPU 和 CPU 支持，那么tensorflow-gpu的目的是什么？

如果它的目的只是支持 GPU（而不是 CPU），那么在尝试运行我的兼容性代码时它怎么会失败？它的使用是否有任何限制，例如关于tensorflow包装？

PS 我已经重新安装了 tensorflow 2.5.0，现在我的兼容模式又可以正常工作了。

标签： tensorflow

解决方案

推荐阅读