推荐用于Tensorflow的GPU

时间:2016-03-31 18:57:26

标签: gpu tensorflow

据我所知,Tensorflow需要(用于GPU计算)具有Nvidia Compute Capability> = 3.0的GPU。有许多这样的GPU可供选择。面向游戏的GPU,例如GeForce模型比面向计算的模型便宜得多,例如特斯拉。我有限的承诺是,面向计算的模型可能缺少视频输出(不需要计算),游戏模型可能正在进行32位数学而不是64位。假设Tensorflow使用(或更喜欢)64位,这样做是否意味着如果与Tensorflow一起使用游戏模型将无法工作或将产生不足的结果?在选择要与Tensorflow一起使用的GPU时,应该寻找哪些属性?

2 个答案:

答案 0 :(得分:4)

支持GPU的TensorFlow版本具有以下要求:

  • 64位Linux
  • Python 2.7
  • NVIDIACUDA®7.5(Pascal GPU所需的CUDA 8.0)
  • NVIDIA cuDNN v4.0(最低)或v5.1(推荐)

TensorFlow GPU支持需要具有NVidia Compute Capability> = 3.0的GPU卡。支持的卡包括但不限于:

  • NVidia Titan
  • NVidia Titan X
  • NVidia K20
  • NVidia K40

您可以看到他们的官方文档Tensorflow GPU support

答案 1 :(得分:2)

游戏GPU可以很好地工作。你想要一个拥有大量内存和CUDA核心的最新GPU。如今,大多数人在GPU上训练神经网络使用32位浮点数。