tensorflow2.0 - 如何在 Ubuntu 20.04 上安装 cuda 11

Question

TensorFlow 官方推荐

所以，我使用的是 Ubuntu 20.4，我想使用 2.3 版的 Tensorflow。官方 Tf 消息称支持 10.1，但我找不到 CUDA 10.1 的 Ubuntu 20.4 安装。是否可以在 Ubuntu 上使用 CUDA 10.1，如果不能，我该如何安装 CUDA 11，这样我才能使 TF 2.3 工作？

标签： tensorflow2.0

Answer 1

是的你可以！通常的方法是从源代码构建 TF，这可能需要很多小时（至少我读到的就是这样）。这是必需的，因为 tensorflow 是使用特定的 cuda 版本编译的，这就是它们必须匹配的原因。经过一番研究我发现，davidenunes 编译了不同的 TF 版本和不同的 cuda 版本，所以你不需要这样做！

看看他的github并拉取你需要的版本。有了这个，我的 2.3 tf 在带有 cuda 11 的 Ubuntu 20.4 上工作。如果你已经准备好工作的 cuda 11 安装，或者你需要 cuda 11，你可以这样做。

但是，您也应该能够在 Ubuntu 20.04 上安装 cuda 10.1，但我建议您使用 cuda 11，因为您可以使用 cudnn 8 来加快学习速度。