Skip to content

Commit

Permalink
adding optuna and plotly to dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
jperez999 committed Nov 20, 2022
1 parent e98bbbc commit 04e4bcb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docker/dockerfile.merlin
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ RUN pip install "cmake<3.25.0" ninja scikit-build
RUN pip install pandas==1.3.5
RUN pip install cupy-cuda117 nvidia-pyindex pybind11 pytest protobuf transformers==4.12 tensorflow-metadata
RUN pip install betterproto cachetools graphviz nvtx scipy sklearn
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
RUN pip install git+https://github.com/rapidsai/asvdb.git@main
# xgboost and treelite are pinned from tritonserver support
RUN pip install xgboost==1.6.2 lightgbm treelite==2.3.0 treelite_runtime==2.3.0
Expand Down

0 comments on commit 04e4bcb

Please sign in to comment.