是否可以将 CUDA 10.2 与 tensorflow 2.1 一起使用？

Question

由于与我的显卡 (GeForce RTX 2060) 不兼容，我发现安装 CUDA 10.1 时出现问题。我成功安装了 CUDA 10.2。但它看起来与 TensorFlow 2.1 不兼容。

当我在 python 上运行这段代码时：

import tensorflow as tf
tf.config.list_physical_devices()

它返回一个空列表，表示没有可用的 GPU。

有没有办法解决这个问题？ 谢谢你

Answer 1

根据此流https://github.com/tensorflow/tensorflow/issues/34759，除非您从源代码构建，否则 tensorflow 2.1 不能与 cuda 10.2 一起使用。

轻松解决问题的唯一方法是降级到 Cuda 10.1，除非您想从源代码构建。