-
Notifications
You must be signed in to change notification settings - Fork 27.5k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Update 3 docker files to use cu118 #23406
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for updating!
The documentation is not available anymore as the PR was closed or merged. |
- name: Cleanup disk | ||
run: | | ||
sudo ls -l /usr/local/lib/ | ||
sudo ls -l /usr/share/ | ||
sudo du -sh /usr/local/lib/ | ||
sudo du -sh /usr/share/ | ||
sudo rm -rf /usr/local/lib/android | ||
sudo rm -rf /usr/share/dotnet | ||
sudo du -sh /usr/local/lib/ | ||
sudo du -sh /usr/share/ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
required to avoid running out of disk space (on runnger)
@@ -35,7 +35,7 @@ RUN [ "$PYTORCH" != "pre" ] && python3 -m pip install --no-cache-dir -U $VERSION | |||
RUN python3 -m pip install --no-cache-dir -U tensorflow==2.12 protobuf==3.20.3 tensorflow_text tensorflow_probability | |||
RUN python3 -m pip uninstall -y flax jax | |||
|
|||
RUN python3 -m pip install --no-cache-dir intel_extension_for_pytorch==$INTEL_TORCH_EXT+cpu -f https://software.intel.com/ipex-whl-stable |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
link starts to fail 2 days ago
@@ -15,6 +15,8 @@ RUN python3 -m pip install --no-cache-dir --upgrade pip | |||
ARG REF=main | |||
RUN git clone https://github.com/huggingface/transformers && cd transformers && git checkout $REF | |||
|
|||
RUN python3 -m pip uninstall -y torch torchvision torchaudio |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
just to make the installation in the following lines work more smoothly without the existing torch stuff
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. |
* fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
* fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
* fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
What does this PR do?
A follow up for #23339.
I will try to build the images and run a small subset of tests to make sure this doesn't break (too many) things before merge.