Workflows still fail due to space limitations in GH Actions. Trying t…

…o build from pytorch runtime image (2.76GB) instead of devel (6.59GB).
compSPI · Aug 22, 2023 · 6cbd995 · 6cbd995
1 parent 76d5462
commit 6cbd995
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,5 @@
 # Use an official pytorch runtime as a parent image
-FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
+FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-runtime
 
 MAINTAINER Youssef Nashed "[email protected]"