From 43e452cdd19b159b04016d2acb986f3e52f7ff5e Mon Sep 17 00:00:00 2001
From: hyh <2667004988@qq.com>
Date: Sun, 17 Mar 2024 11:54:06 +0100
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=94=B9gpus?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 gpu-dockerfile | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/gpu-dockerfile b/gpu-dockerfile
index 831a9ca..7759f60 100644
--- a/gpu-dockerfile
+++ b/gpu-dockerfile
@@ -18,12 +18,8 @@ RUN apt-get update && apt-get install -y libicu-dev wget && \
   tar -xzvf /q/qli-Client-${version}-Linux-x64.tar.gz -C /q/ && \ 
   rm /q/qli-Client-${version}-Linux-x64.tar.gz && \   
   mv /q/qli-Client /q/trainer 
-RUN  wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
-  sudo dpkg -i cuda-keyring_1.1-1_all.deb \
-  sudo apt update -y \ 
-  sudo apt -y install cuda-toolkit \ 
-  export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}\
-  rm cuda-keyring_1.1-1_all.deb
+RUN  wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb 
+RUN sudo dpkg -i cuda-keyring_1.1-1_all.deb  && sudo apt update -y && sudo apt -y install cuda-toolkit && export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} && rm cuda-keyring_1.1-1_all.deb
 
 # 设置工作目录
 WORKDIR /q