网上随便搜一下就会发现关于Tensorflow-gpu的安装文章非常的多,但是写的都比较简略。并且官网的文档写的也比较的简略,并且google 官网上文档对于windows版本的也非常简略。
官网列出的硬件软件需求如下:
硬件要求
系统支持以下支持 GPU 的设备:
- CUDA® 计算能力为 3.5 或更高的 NVIDIA® GPU 卡。请参阅支持 CUDA 的 GPU 卡列表。
软件要求
必须在系统中安装以下 NVIDIA® 软件:
-
NVIDIA® GPU 驱动程序 – CUDA 9.0 需要 384.x 或更高版本。
-
CUDA® 工具包 – TensorFlow 支持 CUDA 9.0。
-
CUDA 工具包附带的 CUPTI。
-
cuDNN SDK(7.2 及更高版本)
-
(可选)NCCL 2.2,可实现多 GPU 支持。
-
(可选)TensorRT 4.0,可缩短在某些模型上进行推断的延迟并提高吞吐量。
除此之外就没有更多的信息了,在官方的pip安装说明页面中可以看到windows版本的其实对于python是有要求的,官方支持的版本如下: