- cont

Lightning-AI · Dec 16, 2021 · 3251248 · 3251248
1 parent 830e3e8
commit 3251248
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/.azure/testing-template.yml b/.azure/testing-template.yml
@@ -39,12 +39,12 @@ jobs:
 
     pool: gridai-spot-pool
     # this need to have installed docker in the base image...
-    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"
-      # image: "pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime"
-      options: "-it --rm --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --shm-size=32g"
+    #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"
+    #  # image: "pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime"
+    #  options: "-it --rm --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all --shm-size=32g"
     steps:
 
     - bash: |