nvcr

Lightning-AI · Dec 16, 2021 · 830e3e8 · 830e3e8
1 parent c36aa73
commit 830e3e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.azure/testing-template.yml b/.azure/testing-template.yml
@@ -42,8 +42,8 @@ jobs:
     container:
       # base ML image: mcr.microsoft.com/azureml/openmpi3.1.2-cuda10.2-cudnn8-ubuntu18.04
       # image: "pytorchlightning/pytorch_lightning:base-cuda-py3.9-torch1.8"
-      # image: "nvcr.io/nvidia/pytorch:21.11-py3-py3"
-      image: "pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime"
+      image: "nvcr.io/nvidia/pytorch:21.11-py3"
+      # image: "pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime"
       options: "-it --rm --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --shm-size=32g"
     steps: