From 1c4a9acc7319221643555c0e8ff1fda2f758c400 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 16 Mar 2023 05:52:40 +0100 Subject: [PATCH] Fix DeepSpeed CI (#22194) * Deal with torch-tensorrt --------- Co-authored-by: ydshieh --- docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile b/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile index 1c20ee261b..2fa1317f82 100644 --- a/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile +++ b/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile @@ -24,9 +24,8 @@ RUN python3 -m pip install --no-cache-dir -U torch torchvision torchaudio --inde RUN python3 -m pip install --no-cache-dir ./transformers[deepspeed-testing] -# This will uninstall torch 2.0.0 -# TODO: uncomment the following line once `torch-tensorrt` is ready for `torch 2.0.0` -# RUN python3 -m pip install torch-tensorrt==1.3.0 --find-links https://github.com/pytorch/TensorRT/releases/expanded_assets/v1.3.0 +# Uninstall `torch-tensorrt` shipped with the base image +RUN python3 -m pip uninstall -y torch-tensorrt # recompile apex RUN python3 -m pip uninstall -y apex