CUDA 11.8 не распознается Tensorflow, но Pytorch распознает ⇐ Python

[Anonymous]

В настоящее время я работаю над проектом машинного обучения, используя Ubuntu 23 с графическим процессором RTX 3060 Ti. Я успешно установил CUDA 11.8, и PyTorch работает отлично, распознает и использует графический процессор без каких-либо проблем.
Однако у меня возникли трудности с TensorFlow. Несмотря на установку пакета tensorflow[and-cuda], TensorFlow, похоже, не распознает графический процессор. Я проверил и подтвердил, что набор инструментов CUDA и cuDNN установлены правильно:

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

2023-11-19 10:04:21.667041: I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2023-11-19 10:04:22.398453: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
2023-11-19 10:04:23.228798: I tensorflow/compiler/xla/stream_executor/cuda/cuda_gpu_executor.cc:995] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero. See more at https://github.com/torvalds/linux/blob/v6.0/Documentation/ABI/testing/sysfs-bus-pci#L344-L355
2023-11-19 10:04:23.257218: W tensorflow/core/common_runtime/gpu/gpu_device.cc:1960] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...

Когда я пытаюсь:

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

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

возвращает только:

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

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 5180605997013099642
xla_global_id: -1
]

Вот соответствующие подробности:

• Операционная система: Ubuntu 23
• Графический процессор: RTX 3060 Ti
• Версия CUDA: 11.8

PyTorch работает без сбоев с поддержкой графического процессора, но TensorFlow отказывается признавать наличие графического процессора.
Я выполнил обычные шаги установки, включая установку набора инструментов CUDA, cuDNN и пакета TensorFlow с поддержкой графического процессора. Однако когда я запускаю код TensorFlow, по умолчанию он выполняется на процессоре.
Будем очень признательны за любые идеи или рекомендации по решению этой проблемы. Если есть конкретные шаги или настройки для TensorFlow в Ubuntu 23 с RTX 3060 Ti, которые мне могут не хватать, дайте мне знать.

Подробнее здесь: https://stackoverflow.com/questions/775 ... torch-does