Skip to content

Commit a14419f

Browse files
committed
update docker image pytorch version for tests
Signed-off-by: Ye Yu <yeyu@nvidia.com>
1 parent 759bb87 commit a14419f

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

.github/workflows/example_tests.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -86,14 +86,14 @@ jobs:
8686
pip_install_extras: "[hf,dev-test]"
8787
runner: linux-amd64-gpu-h100-latest-2
8888

89-
##### Speculative Decoding Example Tests (requires 25.08 image) #####
89+
##### Speculative Decoding Example Tests (requires 26.01 image) #####
9090
speculative-decoding-pr:
9191
needs: [check-file-changes, wait-checks]
9292
if: startsWith(github.ref, 'refs/heads/pull-request/') && needs.check-file-changes.outputs.any_changed == 'true'
9393
uses: ./.github/workflows/_example_tests_runner.yml
9494
secrets: inherit
9595
with:
96-
docker_image: "nvcr.io/nvidia/pytorch:25.08-py3"
96+
docker_image: "nvcr.io/nvidia/pytorch:26.01-py3"
9797
example: speculative_decoding
9898
pip_install_extras: "[hf,dev-test]"
9999
runner: linux-amd64-gpu-l4-latest-1
@@ -103,7 +103,7 @@ jobs:
103103
uses: ./.github/workflows/_example_tests_runner.yml
104104
secrets: inherit
105105
with:
106-
docker_image: "nvcr.io/nvidia/pytorch:25.08-py3"
106+
docker_image: "nvcr.io/nvidia/pytorch:26.01-py3"
107107
example: speculative_decoding
108108
pip_install_extras: "[hf,dev-test]"
109109
runner: linux-amd64-gpu-h100-latest-2

0 commit comments

Comments
 (0)