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 e98bbbc commit 04e4bcbCopy full SHA for 04e4bcb
docker/dockerfile.merlin
@@ -95,7 +95,7 @@ RUN pip install "cmake<3.25.0" ninja scikit-build
95
RUN pip install pandas==1.3.5
96
RUN pip install cupy-cuda117 nvidia-pyindex pybind11 pytest protobuf transformers==4.12 tensorflow-metadata
97
RUN pip install betterproto cachetools graphviz nvtx scipy sklearn
98
-RUN pip install tritonclient[all] grpcio-channelz fiddle wandb npy-append-array
+RUN pip install tritonclient[all] grpcio-channelz fiddle wandb npy-append-array optuna plotly
99
RUN pip install git+https://github.com/rapidsai/asvdb.git@main
100
# xgboost and treelite are pinned from tritonserver support
101
RUN pip install xgboost==1.6.2 lightgbm treelite==2.3.0 treelite_runtime==2.3.0
0 commit comments