CI: update torch version from 1.6 to 1.8 (#784)

Lightning-Universe · Dec 18, 2021 · 8d3fbf7 · 8d3fbf7
1 parent 73945c6
commit 8d3fbf7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/azure-pipelines.yml b/azure-pipelines.yml
@@ -28,7 +28,7 @@ jobs:
     # ToDo: this need to have installed docker in the base image...
     #container: "pytorchlightning/pytorch_lightning:base-cuda-py$[ variables['python.version'] ]-torch1.6"
     container:
-      image: "pytorchlightning/pytorch_lightning:base-cuda-py3.7-torch1.6"
+      image: "pytorchlightning/pytorch_lightning:base-cuda-py3.7-torch1.8"
       #endpoint: azureContainerRegistryConnection
       options: "--runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all"