anchor tf version to avoid errors in 2.10.0 libnvinfer look ups (#625)

NVIDIA-Merlin · Sep 15, 2022 · e348633 · e348633
1 parent 9013783
commit e348633
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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