Skip to content

Commit ccd96d5

Browse files
authored
Pull request/1085 (#1086)
* Fix lightfm 'Illegal instruction' error * Fix lightfm 'Illegal instruction' error
1 parent 4b67f13 commit ccd96d5

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

docker/dockerfile.merlin

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ RUN pip install --no-cache-dir --upgrade pip; pip install --no-cache-dir "cmake<
102102
tritonclient[all] grpcio-channelz fiddle wandb npy-append-array \
103103
git+https://github.com/rapidsai/asvdb.git@main \
104104
xgboost==1.6.2 lightgbm \
105-
lightfm implicit \
105+
implicit \
106106
numba "cuda-python>=11.5,<12.0" fsspec==2022.5.0 llvmlite \
107107
pynvml==11.4.1
108108
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
299299
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/
300300
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/
301301

302-
RUN pip install --no-cache-dir jupyterlab notebook pydot testbook numpy==1.22.4
302+
# There 'Illegal instruction' error, add env 'LIGHTFM_NO_CFLAGS' to workaround
303+
ENV LIGHTFM_NO_CFLAGS=1
304+
RUN pip install --no-cache-dir jupyterlab notebook pydot testbook numpy==1.22.4 lightfm
303305

304306
ENV JUPYTER_CONFIG_DIR=/tmp/.jupyter
305307
ENV JUPYTER_DATA_DIR=/tmp/.jupyter

0 commit comments

Comments
 (0)