Skip to content

GitLab

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

Fix `transformers`' DeepSpeed CI job (#23463) 



* fix

* fix

---------
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 2aa0cc2c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile
View file @ db4d7652
# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel_22-08.html#rel_22-08 # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel-22-12.html#rel-22-12
FROM nvcr.io/nvidia/pytorch:22.12-py3 FROM nvcr.io/nvidia/pytorch:22.12-py3
LABEL maintainer="Hugging Face" LABEL maintainer="Hugging Face"
...@@ -26,6 +26,9 @@ RUN python3 -m pip install --no-cache-dir ./transformers[deepspeed-testing] ...@@ -26,6 +26,9 @@ RUN python3 -m pip install --no-cache-dir ./transformers[deepspeed-testing]
RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/accelerate@main#egg=accelerate RUN python3 -m pip install --no-cache-dir git+https://github.com/huggingface/accelerate@main#egg=accelerate
# Uninstall `transformer-engine` shipped with the base image
RUN python3 -m pip uninstall -y transformer-engine
# Uninstall `torch-tensorrt` shipped with the base image # Uninstall `torch-tensorrt` shipped with the base image
RUN python3 -m pip uninstall -y torch-tensorrt RUN python3 -m pip uninstall -y torch-tensorrt
......
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