如何在 Ubuntu 20.04 上安装 cuda 11

Question

Tensorflow official recommendation

所以，我使用的是 Ubuntu 20.4，我想使用 2.3 版的 Tensorflow。官方 Tf 来源说支持 10.1，但我找不到 Ubuntu 20.4 的 CUDA 10.1 安装。 是否可以在 Ubuntu 上使用 CUDA 10.1，如果不能，我如何安装 CUDA 11，以便让 TF 2.3 工作？

Answer 1

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

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

但是，你应该也可以在Ubuntu 20.04上安装cuda 10.1，但我建议你使用cuda 11，因为你可以使用cudnn 8 speeds 学习了很多。