We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 37b1ac4 commit 6e2db64Copy full SHA for 6e2db64
1 file changed
docker/dockerfile.merlin
@@ -89,7 +89,8 @@ RUN pip install cupy-cuda117 nvidia-pyindex pybind11 pytest protobuf transformer
89
RUN pip install betterproto cachetools graphviz nvtx scipy sklearn
90
RUN pip install tritonclient[all] grpcio-channelz fiddle wandb
91
RUN pip install git+https://github.com/rapidsai/asvdb.git@main
92
-RUN pip install xgboost lightgbm treelite==2.3.0 treelite_runtime==2.3.0
+# xgboost and treelite are pinned from tritonserver support
93
+RUN pip install xgboost==1.6.2 lightgbm treelite==2.3.0 treelite_runtime==2.3.0
94
RUN pip install lightfm implicit
95
RUN pip install numba
96
RUN pip install "cuda-python>=11.5,<12.0"
0 commit comments