GPU与Tensorflow安装的兼容性

Question

深度学习领域的新手。

我需要安装具有GPU支持的Tensorflow。在购买GPU之前，我需要知道GPU是否与Tensorflow兼容。在Tensorflow安装页面中，下面是带有软件的Tensorflow with GPU选项。

您的系统上必须安装以下NVIDIA®软件：

1. NVIDIA®GPU驱动程序— CUDA 9.0需要384.x或更高版本。 CUDA®
2. 工具包— TensorFlow支持CUDA 9.0。 CUPTI随CUDA一起运送
3. 工具包。 cuDNN SDK（> = 7.2）
4. （可选）NCCL 2.2，用于支持多个GPU。
5. （可选）TensorRT 4.0可以改善延迟和吞吐量以进行推理 一些模型。

如果我使用英伟达市场以外的其他品牌，例如技嘉，华硕等，可以说 AORUSGeForce®GTX 1080 Ti Xtreme Edition 11G ，它是否允许我继续使用Tensorflow？

Answer 1

与Nvidia驱动程序兼容 访问网站：https://www.nvidia.com/object/unix.html 单击您感兴趣的驱动程序。在您的情况下为390.87和410.73，单击“支持的产品”。如果有您的GPU，您就可以了。

在您的情况下：是的，您可以使用最新版本的tensorflow。

安装可能有点棘手，但是在互联网上有很多如何安装Nvidia驱动程序，Cuda，cuDNN，NCCL等