在2.4版本的Tensorflow中，pip安装tensorflow后是否应该单独做tensorflow-gpu？

Question

我在安装过程中读到，对于 2.4 版本的 tensorflow，在安装 nvidia cuda 工具包和 cuDNN 之后，执行“$ pip install tensorflow”（访问 GPU）就足够了，而现在没有必要这样做“$ pip install tensorflow-gpu”。

我说得对吗？如果不执行“$ pip install tensorflow-gpu”，我们是否会遗漏任何功能？（附上下面的图片供您参考）

https://towardsdatascience.com/installing-tensorflow-gpu-in-ubuntu-20-04-4ee3ca4cb75d

Answer 1

如果您已经安装了 cuDNN 和 NVIDIA 工具包，那么您可以使用pip install tensorflow。来自 TensorFlow 文档 (https://www.tensorflow.org/install/gpu)：

TensorFlow pip 包包括对支持 CUDA® 的卡的 GPU 支持：

$ pip 安装张量流

本指南介绍了最新稳定版 TensorFlow 的 GPU 支持和安装步骤。

TensorFlow 的旧版本 对于 1.15 及更早的版本，CPU 和 GPU 包是分开的：

$ pip install tensorflow==1.15 # CPU
$ pip install tensorflow-gpu==1.15 # GPU