import tensorflow as tf

print(f"Версия TensorFlow: {tf.__version__}")

# Получаем список физических устройств GPU
gpus = tf.config.list_physical_devices('GPU')

if gpus:
    print(f"✅ Успех! TensorFlow видит {len(gpus)} GPU:")
    for gpu in gpus:
        print(f"   - {gpu}")
    
    # Дополнительная проверка: может ли TF выделить память на GPU
    try:
        tf.config.set_visible_devices(gpus[0], 'GPU')
        print("✅ GPU успешно инициализирован и готов к работе!")
    except RuntimeError as e:
        print(f"❌ Ошибка инициализации GPU: {e}")
else:
    print("❌ TensorFlow НЕ видит GPU. Будет использоваться только CPU.")
    print("💡 Совет: Если у вас Windows, попробуйте установить: pip install tensorflow==2.10.0")