From 3bdf5e856380cd35ece35031f173abd66ef15562 Mon Sep 17 00:00:00 2001
From: Jannik Straube <info@jannik-straube.de>
Date: Fri, 28 Feb 2025 17:02:53 +0100
Subject: [PATCH] fix base image to support nvcc out of the box

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 8dc6460..e5460f0 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.4.1-cuda12.4-cudnn9-runtime
+FROM pytorch/pytorch:2.4.1-cuda12.4-cudnn9-devel
 LABEL maintainer="prime genesys"
 LABEL repository="genesys"