From 31b3bbb37cb17b34c0bf7bd78b5867ba9d8d5651 Mon Sep 17 00:00:00 2001
From: technillogue <technillogue@gmail.com>
Date: Tue, 25 Jul 2023 14:58:58 -0400
Subject: [PATCH] Small fix: don't install python when using python base image

Signed-off-by: technillogue <technillogue@gmail.com>
---
 pkg/dockerfile/generator.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pkg/dockerfile/generator.go b/pkg/dockerfile/generator.go
index 7c5d3adedc..b2ac705206 100644
--- a/pkg/dockerfile/generator.go
+++ b/pkg/dockerfile/generator.go
@@ -97,7 +97,7 @@ func (g *Generator) GenerateBase() (string, error) {
 		return "", err
 	}
 	installPython := ""
-	if g.Config.Build.GPU {
+	if g.Config.Build.GPU && g.useCudaBaseImage {
 		installPython, err = g.installPythonCUDA()
 		if err != nil {
 			return "", err