Initial commit: project structure, scripts and documentation
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
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")
|
||||
Reference in New Issue
Block a user