fix cu11 docker image build (#4165)

Author: CUHKSZzxy

docker/Dockerfile

@@ -69,7 +69,6 @@ COPY docker/install.sh /tmp/install.sh
 RUN --mount=type=cache,target=/root/.cache \
     --mount=type=cache,target=/wheels,from=builder,source=/wheels \
     --mount=type=cache,target=/nccl,from=builder,source=/nccl \
-    --mount=type=cache,target=/debs,from=builder,source=/debs \
     /tmp/install.sh
 
 # explicitly set ptxas path for triton

docker/install.sh

@@ -36,8 +36,8 @@ apt-get clean -y
 rm -rf /var/lib/apt/lists/*
 
 # install GDRCopy debs
-if [ -d "/debs" ] && [ "$(ls -A /debs/*.deb 2>/dev/null)" ]; then
-    dpkg -i /debs/*.deb
+if [ "$(ls -A /wheels/*.deb 2>/dev/null)" ]; then
+    dpkg -i /wheels/*.deb
 fi
 
 # install python packages

docker/prepare_wheel.sh

@@ -53,6 +53,5 @@ if [[ "${CUDA_VERSION_SHORT}" != "cu118" ]]; then
     && tar -xzf v${GDRCOPY_VERSION}.tar.gz && rm v${GDRCOPY_VERSION}.tar.gz \
     && cd gdrcopy-${GDRCOPY_VERSION}/packages \
     && CUDA=/usr/local/cuda ./build-deb-packages.sh \
-    && mkdir -p /debs \
-    && mv ./*.deb /debs/
+    && mv ./*.deb /wheels
 fi