如何安装tensorflow-gpu 2.0版本

Question

我在一个环境中安装了 tensorflow-gpu 2.0 版并且一切正常。 由于某种原因它停止工作。 我创建了一个新环境并执行了conda install tensforflow-gpu。 它安装了 2.4 开发版。 这个版本有各种警告和缺失的模块。 如何安装 tensorflow-gpu 2.0 版？

Answer 1

如果你正在使用 virtulenv，

你可以使用，

conda install tensforflow-gpu=2.0

否则，您可以通过 pip 在 venv 中安装它，

tensorflow-gpu==2.20

更多信息可以在这里找到https://www.tensorflow.org/install/pip