TensorFlow не обнаруживает графический процессор NVIDIA (RTX 3050, CUDA 12.7, TF 2.20.0) ⇐ Python

[Anonymous]

Я пытался заставить TensorFlow использовать мой графический процессор в Windows, и хотя кажется, что все установлено правильно, он показывает 0 доступных графических процессоров.
Настройка системы

• Windows 11
• RTX 3050 GPU для ноутбука
• Драйвер NVIDIA 566.36 (CUDA) 12.7)
• Anaconda3 (Python 3.13)
• TensorFlow 2.20.0

Вот что я сделал на данный момент:

• Установил TensorFlow:

  Код: Выделить всё

  pip install tensorflow==2.20.0
  

• Попробовал новые дополнительные возможности графического процессора, но это не удалось из-за зависимости nvidia-nccl-cu12.

  Код: Выделить всё

  pip install tensorflow[and-cuda] --upgrade
  

  Результат: «Для nvidia-nccl-cu12 не найдено подходящего дистрибутива».
• Поэтому я вручную установил колеса CUDA и cuDNN:

  Код: Выделить всё

  pip install --upgrade nvidia-cublas-cu12 nvidia-cuda-runtime-cu12 nvidia-cudnn-cu12 nvidia-cufft-cu12 nvidia-curand-cu12 nvidia-cusolver-cu12 nvidia-cusparse-cu12
  

  Все успешно установлено.
• Проверено, что CUDA работает:

  Код: Выделить всё

  nvidia-smi
  

  Вывод выглядит нормально: Версия драйвера NVIDIA-SMI 566.36: 566.36 Версия CUDA: 12.7
• Также протестировал драйвер напрямую:

  Код: Выделить всё

  py -c "import ctypes; ctypes.WinDLL('nvcuda.dll'); print('CUDA driver found!')"
  

  Работает нормально: "Драйвер CUDA найден!"
• Затем я проверил TensorFlow:

  Код: Выделить всё

  py -c "import tensorflow as tf; print('TF version:', tf.__version__);   print('GPUs:', tf.config.list_physical_devices('GPU'))"
  

  Вывод: версия TF: 2.20.0 Графические процессоры: []

Итак, графический процессор явно присутствует, CUDA и cuDNN установлены, но TensorFlow все еще не обнаруживает его.
Из того, что я прочитал, это может быть связано с тем, что TensorFlow 2.20.0 в Windows + Python 3.13 не имеет колесо с поддержкой графического процессора еще нет. Все остальное (PyTorch, инструменты CUDA) работает нормально, но TF просто не видит графический процессор.
Кому-нибудь уже удалось заставить графический процессор TensorFlow работать на Python 3.13 с CUDA 12.7?
Или мне следует перейти на Python 3.10/3.11 и вместо этого использовать TensorFlow 2.17.0?

Подробнее здесь: https://stackoverflow.com/questions/798 ... -tf-2-20-0