安装TensorFlow

pip 软件包

TensorFlow pip 软件包对采用 CUDA 的显卡提供 GPU 支持：

pip install tensorflow

必须在系统中安装以下 NVIDIA软件：

• NVIDIA GPU 驱动程序 - CUDA 11.2 要求 450.80.02 或更高版本
• CUDA 工具包：TensorFlow 支持 CUDA 11.2（TensorFlow 2.5.0 及更高版本）
• CUDA 工具包附带的 CUPTI
• cuDNN SDK 8.1.0 ；cuDNN 版本
• （可选）TensorRT 6.0，可缩短用某些模型进行推断的延迟时间并提高吞吐量

GPU 支持官方文档地址：https://www.tensorflow.org/install/gpu