python - 如何安装 tensorflow-gpu 2.0 版

Question

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

标签： pythontensorflow

Answer 1

如果您使用的是virtulenv，

您可以使用，

conda install tensforflow-gpu=2.0

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

tensorflow-gpu==2.20

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