works with deepctr[gpu] on VM instance, but won't work on local computer without NVIDIA smi installed!
latest-gpu-py3-deepctr-stable works on VM, without pip3 install keras installed, but does run deepfm.py and does load model
from tensorflow.python.keras.models import save_model, load_model
from deepctr.layers import custom_objects
model = load_model('./DeepFM.h5', custom_objects)
Content type
Image
Digest
Size
3 GB
Last updated
about 7 years ago
docker pull gchen0119/python-tensorflow:latest-gpu-py3-deepctr-0.6.0-tensorflow-tensorboard-keras-mlflow-stable