21 lines
955 B
Python
21 lines
955 B
Python
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") |