badge

Borda · Borda · commit db49fa98c788 · 2020-10-01T12:29:03.000+02:00
diff --git a/.drone.yml b/.drone.yml
@@ -20,7 +20,7 @@ name: torch-GPU
 
 steps:
 - name: testing
-  image: pytorchlightning/pytorch_lightning:base-cuda-py3.7-torch1.5
+  image: pytorchlightning/pytorch_lightning:base-cuda-py3.7-torch1.6
 
   environment:
     SLURM_LOCALID: 0
