我可以安装 cuda 10.2 以使用 tensorflow 2.1 还是必须是 cuda 10.1？答案

【问题标题】：Can I install cuda 10.2 for using tensorflow 2.1 or it has to be cuda 10.1?我可以安装 cuda 10.2 以使用 tensorflow 2.1 还是必须是 cuda 10.1？
【发布时间】：2021-10-22 17:58:43
【问题描述】：

我可以安装 cuda 10.2 以使用 tensorflow 2.1 还是必须是 cuda 10.1？我使用的是 ubuntu 18.04，并且我有一个 NVIDIA Quadro P5000。

【问题讨论】：

TensorFlow documentation 似乎很明确：“TensorFlow 支持 CUDA 10.1 (TensorFlow >= 2.1.0)”
在大多数情况下，CUDA 10.2 不能用作与 CUDA 10.1 链接的应用程序的替代方案。如果您安装 TF 二进制文件（而不是从源代码构建），那么您通常需要提供与二进制文件链接的 CUDA 版本。这些 cmets 大约适用于任何 2 个不同的 CUDA 版本，适用于与 CUDA 链接的任何应用程序。
我曾尝试将 CUDA 10.1 用于 TF 2.1，但没有成功。您想使用 CUDA 10.2 的任何特殊原因？
@thushv89 这对我来说只是一个问题，但似乎不可能做到。
我可以使用 CUDA 10.1 成功安装 tensorflow 2.1。

标签： tensorflow ubuntu

【解决方案1】：

在此处（回答部分）提供解决方案，即使它出现在评论部分，也是为了社区的利益。

不，根据 Tensorflow 文档，TensorFlow 支持CUDA 10.1 (TensorFlow >= 2.1.0)，请参考兼容版本details

【讨论】：

【解决方案2】：

Pytorch 需要 CUDA 10.2，但 Tensorflow 需要 cuda 10.1。是开玩笑吗？

不，您可以将 cuda 版本 10.2 与 tensorflow 2.0 一起使用。这很简单。

为什么：

运行“import tensorflow”时，tensorflow 将在 LD_LIBRARY_PATH 中搜索名为“libcudart.so.$.$”的库。对于带有 cuda 10.1 的 tensorflow 2.1.0-2.3.0，它是“libcudart.so.10.1”。使用 cuda 10.2，我们没有 'libcudart.so.10.1'，所以会出现错误。

其实cuda 10.1和cuda 10.2没有任何区别，所以我们可以通过软链接解决这个问题。

如何

cd /usr/local/cuda-10.2/targets/x86_64-linux/lib/
ln -s libcudart.so.10.2.89 libcudart.so.10.1

/usr/local/cuda-10.2/extras/CUPTI/lib64
ln -s libcupti.so.10.2.75 libcupti.so.10.1

cd /usr/local/cuda-10.2/lib64
ln -s libcudnn.so.8 libcudnn.so.7

vim /etc/profile
导出 CUDA_HOME=/usr/local/cuda-10.2
导出 LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${CUDA_HOME}/extras/CUPTI/lib64
导出 PATH=${CUDA_HOME}/bin:${PATH}
来源 /etc/profile

点击按钮查看图片。 Done!

【讨论】：