Pull request/1085 (#1086)

* Fix lightfm 'Illegal instruction' error * Fix lightfm 'Illegal instruction' error
NVIDIA-Merlin · Dec 15, 2023 · ccd96d5 · ccd96d5
1 parent 4b67f13
commit ccd96d5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin
@@ -102,7 +102,7 @@ RUN pip install --no-cache-dir --upgrade pip; pip install --no-cache-dir "cmake<
                 tritonclient[all] grpcio-channelz fiddle wandb npy-append-array \
                 git+https://github.com/rapidsai/asvdb.git@main \
                 xgboost==1.6.2 lightgbm \
-                lightfm implicit \
+                implicit \
                 numba "cuda-python>=11.5,<12.0" fsspec==2022.5.0 llvmlite \
                 pynvml==11.4.1
 RUN pip install --no-cache-dir treelite==2.4.0 treelite_runtime==2.4.0
@@ -299,7 +299,9 @@ COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python${PYTHON_VERSION}/dist-p
 COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python${PYTHON_VERSION}/dist-packages/numba-*.dist-info /usr/local/lib/python${PYTHON_VERSION}/dist-packages/numba.dist-info/
 COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python${PYTHON_VERSION}/dist-packages/cubinlinker-*.dist-info /usr/local/lib/python${PYTHON_VERSION}/dist-packages/cubinlinker.dist-info/
 
-RUN pip install --no-cache-dir jupyterlab notebook pydot testbook numpy==1.22.4
+# There 'Illegal instruction' error, add env 'LIGHTFM_NO_CFLAGS' to workaround
+ENV LIGHTFM_NO_CFLAGS=1
+RUN pip install --no-cache-dir jupyterlab notebook pydot testbook numpy==1.22.4 lightfm
 
 ENV JUPYTER_CONFIG_DIR=/tmp/.jupyter
 ENV JUPYTER_DATA_DIR=/tmp/.jupyter