NVIDIA Merlin is an open source library providing end-to-end GPU-accelerated recommender systems, from feature engineering and preprocessing to training deep learning models and running inference in production.
Apache License 2.0
758
stars
113
forks
source link
Install pytorch lightning in torch container #1010
The new Models pytorch backend requires pytorch-lightning and lightning-utilities, which were added in Models/#1126.
This PR installs the new dependencies in the PyTorch container. The change in size is minimal -- uncompressed image size is 17.8 GB both before and after installing these packages.
The new Models pytorch backend requires
pytorch-lightning
andlightning-utilities
, which were added in Models/#1126.This PR installs the new dependencies in the PyTorch container. The change in size is minimal -- uncompressed image size is 17.8 GB both before and after installing these packages.