带有GPU和CUDA v5.5的TensorFlow



我想使用启用的tensorflow版本。我读到它最适合CUDA 7.5,但是我使用的服务器已安装了CUDA版本5.5。

我可以用CUDA 5.5配置TensorFlow吗?如果是,怎么样?我已经在虚拟环境中安装了TensorFlow。

来自安装文档:

为了使用GPU支持构建或运行TensorFlow,NVIDIA的CUDA工具包(> = 7.0)和CUDNN(> = V2)都需要安装。

如果您不从源安装,则需要使用7.5版。


还确保您具有兼容的GPU:

TensorFlow GPU支持需要具有NVIDIA COMPUTE功能的GPU卡> = 3.0。受支持的卡包括但不限于:

  • Nvidia Titan
  • Nvidia Titan X
  • nvidia k20
  • nvidia k40

最新更新