We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9e419aa commit 14c2dc1Copy full SHA for 14c2dc1
.azure/gpu-benchmarks.yml
@@ -46,7 +46,7 @@ jobs:
46
variables:
47
DEVICES: $( python -c 'print("$(Agent.Name)".split("_")[-1])' )
48
container:
49
- image: "pytorchlightning/pytorch_lightning:base-cuda12.16.3-py3.12-torch2.8"
+ image: "pytorchlightning/pytorch_lightning:base-cuda12.6.3-py3.12-torch2.8"
50
options: "--gpus=all --shm-size=32g"
51
strategy:
52
matrix:
0 commit comments