From d7da7ff8de7813155a5f95be01245062d518371b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B5=81=E5=B9=B4?= <williamdeyang@outlook.com>
Date: Wed, 4 Dec 2024 20:52:24 +0800
Subject: [PATCH] Update llama-server-cuda.Dockerfile

---
 .devops/llama-server-cuda.Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.devops/llama-server-cuda.Dockerfile b/.devops/llama-server-cuda.Dockerfile
index bf8a198f9..0f1a9d534 100644
--- a/.devops/llama-server-cuda.Dockerfile
+++ b/.devops/llama-server-cuda.Dockerfile
@@ -1,6 +1,6 @@
-ARG UBUNTU_VERSION=22.04
+ARG UBUNTU_VERSION=20.04
 # This needs to generally match the container host's environment.
-ARG CUDA_VERSION=12.6.0
+ARG CUDA_VERSION=11.3.1
 # Target the CUDA build image
 ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
 # Target the CUDA runtime image