Skip to content

Commit

Permalink
[CI] Promote nvidia docker container to version 24.4 (#354)
Browse files Browse the repository at this point in the history
Promote nvidia docker container to version 24.4 => Getting pytorch 2.3

Regression passed 
https://github.com/CentML/hidet/actions/runs/9964867474
  • Loading branch information
vadiklyutiy authored Jul 17, 2024
1 parent bab24b1 commit 83389be
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM nvcr.io/nvidia/pytorch:23.10-py3
FROM nvcr.io/nvidia/pytorch:24.04-py3
ADD ./hidet /workspace/hidet
ADD ./models /workspace/models
WORKDIR /workspace
Expand All @@ -9,4 +9,4 @@ RUN pip install -r hidet/requirements.txt && \
WHEEL=$(find hidet/scripts/wheel/built_wheel -maxdepth 1 -name '*.whl') && \
pip install --no-deps --force-reinstall $WHEEL && \
pip install -e models && \
hidet cache clear --all
hidet cache clear --all

0 comments on commit 83389be

Please sign in to comment.