From 29035281fc17bb19f8c98707c11bc99a3bab74a2 Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Fri, 4 Jul 2025 10:44:00 +0800
Subject: [PATCH] .

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

diff --git a/Dockerfile b/Dockerfile
index 3486686..89459ac 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -116,7 +116,7 @@ FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF-8
 
-RUN apt-get update && apt-get install -y --no-install-recommends gcc g++ ninja-build \
+RUN apt-get update && apt-get install -y --no-install-recommends gcc g++ ninja-build cuda-toolkit-12-6\
         python3 python3-dev python3-pip python3-distutils ca-certificates \
         libopenblas-dev libgomp1 libcupti-dev libnuma1 libopenmpi-dev openmpi-bin libnuma-dev libpng16-16 libjpeg8 && \
     rm -rf /var/lib/apt/lists/* && \