要想使你的电脑支持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的样例中随便找一个样例项目: