Можно ли использовать CUDA 10.2 с tenorflow 2.1?

Question

У меня проблемы с установкой CUDA 10.1 из-за несовместимости с моей видеокартой c (GeForce RTX 2060). Я успешно установил CUDA 10.2. Но это выглядит несовместимым с Tensorflow 2.1.

Когда я запускаю этот код на python:

import tensorflow as tf
tf.config.list_physical_devices()

, он возвращает пустой список, что означает отсутствие доступных графических процессоров.

Есть ли способ решить эту проблему? Спасибо

Answer 1

Согласно этому потоку https://github.com/tensorflow/tensorflow/issues/34759 тензор потока 2.1 не работает с cuda 10.2, если вы не строите из исходных кодов.

Единственный способ легко решить вашу проблему - перейти на версию Cuda 10.1, если вы не хотите строить из исходного кода.