2016-06-26 185 views
0

我想使用GPU启用版本的Tensorflow。我读过它与CUDA 7.5的最佳配合,但我使用的服务器安装了CUDA 5.5版本。Tensorflow与GPU和CUDA v5.5

我可以用CUDA 5.5配置Tensorflow吗?如果是,如何?我已经在一个虚拟环境中安装了tensorflow。

回答

1

从安装文档:

为了建立或GPU支持运行TensorFlow,无论NVIDIA的CUDA工具包(> = 7.0)和cuDNN(> = V2)需要安装。

如果您不从源代码安装,您将需要使用7.5版本。


另外,还要确保你有一个兼容GPU:

TensorFlow GPU的支持需要具有GPU卡的NVidia计算能力> = 3.0。支持的卡包括但不限于:

  • 的NVidia泰坦
  • 的NVidia泰坦X
  • 的NVidia K20
  • 的NVidia K40