Перенос модели LSTM с процессора на графический процессор ⇐ Python

[Anonymous]

У меня есть очень простая модель LSTM, которую я построил в тензорном потоке, и она работает на процессоре. Однако я хочу использовать эту модель на графическом процессоре. Для pytorch я определил устройство и т. д., однако для тензорного потока я понятия не имею, почему он не может работать. У вас есть какое-нибудь предложение для меня? Спасибо
model = Sequential()
model.add(LSTM(64, activation='relu', input_shape=(X_train.shape[1], X_train.shape[2]), return_sequences=True))
model.add(LSTM(32, activation='relu', return_sequences=False))
model.add(Dropout(0.1))
model.add(Dense(Y_train.shape[1], kernel_regularizer='l2'))
callback = tf.keras.callbacks.EarlyStopping(monitor='loss', patience=50)
opt = keras.optimizers.Adam(learning_rate=0.0008)
model.compile(optimizer=opt, loss='mse')
#model.summary()
history = model.fit(X_train, Y_train, epochs=2, batch_size=100, validation_data=(X_val, Y_val), callbacks=[callback],verbose=1, device).to(device)


Подробнее здесь: https://stackoverflow.com/questions/720 ... cpu-to-gpu