From 0076f1154160f53a6c5de8a3716783071f6ef617 Mon Sep 17 00:00:00 2001
From: Yineng Zhang <me@zhyncs.com>
Date: Tue, 13 Aug 2024 19:08:43 +0800
Subject: [PATCH] fix: use devel for Triton's compiler requirements (#1074)

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

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 95127b33a..2f7541c9a 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,5 +1,5 @@
 ARG CUDA_VERSION=12.1.1
-FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu20.04
+FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
 ARG BUILD_TYPE=all
 ENV DEBIAN_FRONTEND=noninteractive