2.4版本的Tensorflow中,pip安装tensorflow后是否要单独做tensorflow-gpu?

In 2.4 version of Tensorflow, should we do tensorflow-gpu seperately after pip installing tensorflow?

我在安装过程中看到,对于 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

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

The TensorFlow pip package includes GPU support for CUDA®-enabled cards:

$ pip install tensorflow

This guide covers GPU support and installation steps for the latest stable TensorFlow release.

Older versions of TensorFlow For releases 1.15 and older, CPU and GPU packages are separate:

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