bump

Author: Borda

.azure/gpu-benchmarks.yml

@@ -46,7 +46,7 @@ jobs:
     variables:
       DEVICES: $( python -c 'print("$(Agent.Name)".split("_")[-1])' )
     container:
-      image: "pytorchlightning/pytorch_lightning:base-cuda12.1.0-py3.12-torch2.5"
+      image: "pytorchlightning/pytorch_lightning:base-cuda12.16.3-py3.12-torch2.8"
       options: "--gpus=all --shm-size=32g"
     strategy:
       matrix: