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