From de5186908f56a1901bf2179c850105b9bf142862 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B5=81=E5=B9=B4?= Date: Fri, 6 Dec 2024 03:00:41 +0800 Subject: [PATCH] Update llama-server-cuda.Dockerfile --- .devops/llama-server-cuda.Dockerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.devops/llama-server-cuda.Dockerfile b/.devops/llama-server-cuda.Dockerfile index fb52d573d..130bbd633 100644 --- a/.devops/llama-server-cuda.Dockerfile +++ b/.devops/llama-server-cuda.Dockerfile @@ -1,11 +1,11 @@ ARG UBUNTU_VERSION=20.04 # This needs to generally match the container host's environment. -ARG CUDA_VERSION=11.3.1 +ARG CUDA_VERSION=11.4.0 # Target the CUDA build image -ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION} +ARG BASE_CUDA_DEV_CONTAINER=registry.cn-hangzhou.aliyuncs.com/reg_pub/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION} # Target the CUDA runtime image -ARG BASE_CUDA_RUN_CONTAINER=nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION} +ARG BASE_CUDA_RUN_CONTAINER=registry.cn-hangzhou.aliyuncs.com/reg_pub/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION} FROM ${BASE_CUDA_DEV_CONTAINER} AS build ARG DEBIAN_FRONTEND=noninteractive