Relative Content

Tag Archive for pythontensorflowkerascudanvidia

tensorflow stucked in load_model

I’m trying to load a model that I have trained in Google Colab, using Ubuntu 24.04 with conda 24.5.0, NVIDIA-SMI 555.58.02, Driver Version 556.12, CUDA 12.5 All my drivers are updated, but when I try to run the follwing code, it gets stucked loading the model. I have already tried to import the Model function from tensorflow.keras.models and load directly from keras, with the same results. I had to downgrade python from 3.12 to 3.9, otherwise tensorflow wouldn’t work, I also have tried to load other models that are loaded in colab with no problem, with similar results. I have the option to save the model as .keras, but when I do, it gives me Error loading model: SavedModel file does not exist at: model4.keras/{saved_model.pbtxt|saved_model.pb}. Should I try a specific combination of package version for this code to work? Any direction would be welcome.