Skip to content

Commit

Permalink
Torch fix (#207)
Browse files Browse the repository at this point in the history
* fix torch horovod explicit do not need tensorflow

* fix dask-cuda call
  • Loading branch information
jperez999 committed Apr 14, 2022
1 parent 6025a52 commit 6ed50fb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docker/inference/dockerfile.torch
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ RUN pip install cupy-cuda115 nvidia-pyindex pybind11 pytest protobuf transformer
RUN pip install betterproto cachetools graphviz nvtx scipy sklearn
RUN pip install numba --no-deps
RUN pip install tritonclient[all] grpcio-channelz
RUN pip install dask==2021.11.2 distributed==2021.11.2 dask[dataframe]==2021.11.2 dask-cuda22.2.0
RUN pip install dask==2021.11.2 distributed==2021.11.2 dask[dataframe]==2021.11.2 dask-cuda==22.2.0
RUN pip install git+https://github.com/rapidsai/asvdb.git@main
RUN pip install "cuda-python>=11.5,<12.0"

Expand Down

0 comments on commit 6ed50fb

Please sign in to comment.