Tensorflow GPU支持

要想使你的电脑支持GPU运算(只适用NVIDIA)就必须相关的驱动程序。
强烈建议Linux用户： 为了简化安装并避免库冲突，建议您使用支持 GPU 的 TensorFlow Docker 映像。

硬件要求

系统支持以下支持 GPU 的设备：

CUDA® 计算能力为 3.5 或更高的 NVIDIA® GPU 卡。请参阅支持 CUDA 的 GPU 卡列表。

软件要求

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

• NVIDIA® GPU 驱动程序 - CUDA 10.0 需要 410.x 或更高版本。
• CUDA® 工具包 - TensorFlow 支持 CUDA 10.0（TensorFlow 1.13.0 及更高版本）
• CUDA 工具包附带的 CUPTI。
• cuDNN SDK（7.4.1 及更高版本）
• （可选）TensorRT 5.0，可缩短在某些模型上进行推断的延迟并提高吞吐量。

安装步骤

• 对于 NVIDIA® GPU 驱动程序来说，很简单直接装完就行。
• CUDA® 工具包安装也是很简单：当到安装界面时尽量选择精简安装。
• CUDA 工具包附带的 CUPTI，一般来说按照前面的精简安装都会安装CUPTI。
• cuDNN SDK下载完是一个压缩包，只需将其解压到 CUDA® 工具包的目录就行。

验证是否安装成功

可以到CUDA的样例中随便找一个样例项目：