我如何让tensorflow在具有2.x能力的GPU上运行？

Question

我已经成功地在Linux Ubuntu 16.04上安装了tensorflow(GPU)，并做了一些小的修改，以便使它与新的Ubuntu LTS版本一起工作。

Answer 1

tensorflow的最新GPU版本要求计算能力3.5或更高（并使用cuDNN访问GPU）。

cuDNN还需要CC3.0或更高的GPU：

使用Pascal、Kepler、Maxwell、Tegra K1或Tegra X1 GPU的Windows、Linux和MacOS系统支持cuDNN。

null

CUDNN也不支持旧的GPU（例如compute capability 1.x)。

注意，从来没有任何版本的cuDNN或任何版本的TF正式支持低于CC3.0的NVIDIA GPU。cuDNN的初始版本开始时需要CC3.0GPU，TF的初始版本开始时需要CC3.0GPU。