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

时间:2018-08-08 09:43:12

标签: tensorflow cuda

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

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

对于这个愚蠢的问题很抱歉,我是安装DL框架的新手:-)

非常感谢您的帮助!

1 个答案:

答案 0 :(得分:3)

  

在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。 driver 其中ships with CUDA 9.0是384系列,不支持您的GPU。如果您提到驱动程序支持问题,那么解决方案将是为GPU安装推荐的驱动程序,并且仅安装CUDA 9.0捆绑包中的CUDA工具包,而不是默认的工具包和驱动程序。

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