是否可以將 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

According to this stream https://github.com/tensorflow/tensorflow/issues/34759 , tensorflow 2.1 is not working with cuda 10.2 unless you build from the sources.

輕松解決問題的唯一方法是降級到 Cuda 10.1，除非您想從源代碼構建。