如何在Nvidia Quadro GV100上运行tensorflow-gpu？

Question

我目前是在职学生，现在在使用Nvidia Quadro GV100 GPU的计算机上安装Tensorflow-gpu时遇到问题。

在Tensorflow主页上，我发现我需要安装CUDA 9.0和Cudnn 7.x才能运行Tensorflow-gpu 1.9。 问题是我找不到支持GV100的合适CUDA版本。 可能还没有CUDA版本吗？ 是否有可能无法将GV100用于tensoflow-gpu？

对不起这个愚蠢的问题，我是安装DL框架的新手:-)

非常感谢您的帮助！

Answer 1

在Tensorflow主页上，我发现我需要安装CUDA 9.0和Cudnn 7.x才能运行Tensorflow-gpu 1.9。

也就是说，如果您要安装预构建的Tensorflow二进制发行版。 在这种情况下，您需要使用Tensorflow二进制文件所针对的CUDA版本，在这种情况下为CUDA 9.0

问题是我找不到支持GV100的合适CUDA版本

CUDA 9.0和更高版本的工具包完全支持Volta卡，并且应包括Quadro GV100。 CUDA 9.0附带的驱动程序是384系列，不支持您的GPU。 如果您提到驱动程序支持问题，那么解决方案将是为GPU安装推荐的驱动程序，并且仅安装CUDA 9.0捆绑包中的CUDA工具包，而不是默认的工具包和驱动程序。

否则，您可以使用CUDA 9.1或9.2，它们应使用其提供的驱动程序来支持您的GPU，但随后您将需要从源代码自己构建Tensorflow。