安装 Tensorflow

安装 Tensorflow

我的笔记本上有一块 GPU nvidia Geforce 820m。我想在我的 Ubuntu 16.04 上安装 tensorflow。我不确定我应该安装 GPU 还是非 GPU 版本?
它的计算能力为 2.1,但 tensorflow 需要 3.0 以上

答案1

如果计算能力小于 3.0 TF 将无法按照安装指南

使用支持 GPU 的 TF 并不总是更快。如果您拥有快速的多核 CPU,那么在训练小型网络时,您可能无法从 GPU 支持中获益。将数据加载到 GPU 存储中需要时间,比将数据复制到 CPU-RAM 中要长得多。对于深度网络来说,这无关紧要,GPU 会更快。

我有一台配备四核 i7(第 6 代)和 GTX 960M 的笔记本电脑,像谷歌 MNIST 教程中使用的小型网络在使用 GPU 时总体上训练速度较慢,因为必须先加载数据。

另外,安装支持 GPU 的 TF 比安装简单的 TF 花费的时间要多得多pip3 install tensorflow

答案2

有了 gpu 支持相信我会更快

只需确保先为你的卡安装专有驱动程序

相关内容