From 32512484c7f1a5a5870761b2f002025ffffd04fe Mon Sep 17 00:00:00 2001 From: Jirka Date: Fri, 17 Dec 2021 00:13:46 +0100 Subject: [PATCH] - cont --- .azure/testing-template.yml | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/.azure/testing-template.yml b/.azure/testing-template.yml index cef65267..050f3290 100644 --- 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: |