Ошибка ImportError: невозможно импортировать имя «multi_gpu_model» из «tensorflow.keras.utils» в textgenrnn

Question

Я пытаюсь тренироватьtextgenrnnсмоделируйте на Python и сохраните веса. У меня есть текстовый файл со списком заголовков, которые я хочу использовать. это мой код:

      from textgenrnn import textgenrnn
t = textgenrnn()
t.train_from_file(r"filepath goes here", num_epochs=10)

Но я получаю эту ошибку:

      2023-05-25 14:40:27.353193: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
2023-05-25 14:40:27.353897: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
Traceback (most recent call last):
  File "C:\Users\HP\Desktop\train_er.py", line 1, in <module>
    from textgenrnn import textgenrnn
  File "C:\Users\HP\anaconda3\lib\site-packages\textgenrnn\__init__.py", line 1, in <module>
    from .textgenrnn import textgenrnn
  File "C:\Users\HP\anaconda3\lib\site-packages\textgenrnn\textgenrnn.py", line 5, in <module>
    from tensorflow.keras.utils import multi_gpu_model
ImportError: cannot import name 'multi_gpu_model' from 'tensorflow.keras.utils' (C:\Users\HP\anaconda3\lib\site-packages\keras\api\_v2\keras\utils\__init__.py)

У меня естьtensorflow 2.10.0,keras 2.10.0,textgenrnn 2.0.0и CUDA установлен