Skip to content

GitLab

Unverified Commit d994473b authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Uninstall `torch_tensorrt` in `DeepSpeed` CI image for now (#20758) 



Uninstall torch_tensorrt for now
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent ba9da49a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile
View file @ d994473b
...@@ -22,6 +22,9 @@ RUN python3 -m pip install --no-cache-dir -U torch==$PYTORCH torchvision torchau ...@@ -22,6 +22,9 @@ RUN python3 -m pip install --no-cache-dir -U torch==$PYTORCH torchvision torchau
RUN python3 -m pip install --no-cache-dir ./transformers[deepspeed-testing] RUN python3 -m pip install --no-cache-dir ./transformers[deepspeed-testing]
# The base image ships with `torch_tensorrt` which currently causes the DeepSpeed CI fail from test collection
RUN python3 -m pip uninstall -y torch_tensorrt
# Pre-build **latest** DeepSpeed, so it would be ready for testing (otherwise, the 1st deepspeed test will timeout) # Pre-build **latest** DeepSpeed, so it would be ready for testing (otherwise, the 1st deepspeed test will timeout)
RUN python3 -m pip uninstall -y deepspeed RUN python3 -m pip uninstall -y deepspeed
# This has to be run (again) inside the GPU VMs running the tests. # This has to be run (again) inside the GPU VMs running the tests.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment