diff --git a/ci/dockerfile.ci b/ci/dockerfile.ci index 3ac8285e0..0360a8806 100644 --- a/ci/dockerfile.ci +++ b/ci/dockerfile.ci @@ -9,7 +9,7 @@ FROM ${BASE_IMAGE} COPY --chown=1000:1000 --from=triton /opt/tritonserver/backends/tensorflow2 backends/tensorflow2/ COPY --chown=1000:1000 --from=triton /opt/tritonserver/backends/pytorch backends/pytorch/ -RUN pip install tensorflow-gpu +RUN pip install tensorflow-gpu==2.9.2 RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu113 RUN pip install torchmetrics==0.3.2 matplotlib RUN pip install fastai fastcore fastprogress fastdownload --no-deps