Keras-тюнер зависает после произвольного количества попыток использования WSL ⇐ Python

[Anonymous]

Недавно мне пришлось переключиться на использование WSL 2, чтобы включить вычисления на графическом процессоре с помощью Tensorflow. У меня был фрагмент кода, работающий на Windows (ЦП), который настраивает три гиперпараметра с помощью keras-tuner. Однако при переходе на WSL 2 внезапно, после Х испытаний, настройка прекращается:
[img]https://i.sstatic .net/Anhsuw8J.png[/img]

Иногда он зависает после 3 попыток, иногда 2. Он не вылетает, просто остается на эпохе 1/3, поэтому нет сообщения об ошибке. . Это код настройки (который работал на Windows).

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

tuner = kt.Hyperband(build_tune_model,
objective='val_accuracy',
max_epochs=25,
factor=3,
directory='keras_tuner',
project_name='hyperband_tune')

early_stopping = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience=5)
tensorboard_callback = tf.keras.callbacks.TensorBoard("/tmp/tb_logs")

tuner.search(
ds_train,
epochs=400,
validation_data=ds_valid,
steps_per_epoch=train_size // BATCH_SIZE,
validation_steps=valid_size,
callbacks=[early_stopping, tensorboard_callback]
)

Что я пробовал:

• Ожидание в течение часа (предыдущие испытания требовали менее 4 минут)< /li>
  Использование различных тюнеров (байесовская оптимизация и HyperBand)
• Проверка статистики производительности ПК. Кажется, что процессор используется (53%), а графический процессор совершенно не используется (это имеет смысл, поскольку эпоха не начинается). Загрузка ЦП снижается при прерывании работы ядра.

Дополнительная информация:

• Я использую Python 3.11.3.
• Я установил tensorflow с помощью python3.11 -m pip install tensorflow[and-cuda], который не выдал никаких сообщений об ошибках. Установлен
  tensorflow 2.17.0. Перед установкой я удалил все существующие установки CUDA на своем компьютере, на всякий случай.
• Я установил WSL 2 с помощью wsl --install. . Я использую дистрибутив Ubuntu по умолчанию.

Стоит отметить:
При загрузке Tensorflow я получаю следующие предупреждения, которых не было до перехода на WSL 2:
[img]https://i.sstatic .net/663SFMBM.png[/img]

Может ли одно из этих предупреждений быть проблемой? Кроме того, я использую Jupyter Notebook, попробую запустить свой код с терминала и посмотреть, работает ли он.
Буду очень признателен за любую помощь. Если какая-либо информация отсутствует, пожалуйста, дайте мне знать. Заранее спасибо.

Подробнее здесь: https://stackoverflow.com/questions/788 ... -using-wsl