Update tests container

Signed-off-by: ajrasane <[email protected]>

Author: ajrasane

.github/workflows/gpu_tests.yml

@@ -62,7 +62,7 @@ jobs:
     runs-on: linux-amd64-gpu-l4-latest-1
     timeout-minutes: 120
     container: &gpu_container
-      image: nvcr.io/nvidia/pytorch:25.06-py3
+      image: nvcr.io/nvidia/pytorch:25.08-py3
       env:
         GIT_DEPTH: 1000 # For correct version for tests/gpu/torch/quantization/plugins/test_megatron.py
         PIP_CONSTRAINT: "" # Disable pip constraint for upgrading packages

.gitlab/tests.yml

@@ -27,7 +27,7 @@ unit:
 .multi-gpu-tests-default:
   extends: .tests-default
   timeout: 90m
-  image: nvcr.io/nvidia/pytorch:25.06-py3
+  image: nvcr.io/nvidia/pytorch:25.08-py3
   variables:
     GIT_DEPTH: 1000 # For correct version for tests/gpu/torch/quantization/plugins/test_megatron.py
   tags: [docker, linux, 2-gpu]