docker gpu arm update
Some checks failed
build_docker / essential (pull_request) Successful in 0s
build_docker / build_cpu (pull_request) Successful in 7s
build_docker / build_gpu (pull_request) Failing after 7s

This commit is contained in:
2026-01-17 11:04:25 +01:00
parent 78fe3e8c81
commit 9daa496ab6
2 changed files with 21 additions and 5 deletions

View File

@@ -92,8 +92,16 @@ RUN mkdir -p build
WORKDIR /build/Paddle/build
# Configure CMake for ARM64 + CUDA build
# Note: Adjust CUDA_ARCH_NAME based on your GPU architecture
# Common values: Auto, Ampere, Ada, Hopper
#
# CUDA_ARCH is auto-detected from host GPU and passed via docker-compose.
# To detect: nvidia-smi --query-gpu=compute_cap --format=csv,noheader
# Example: 12.1 -> use "90" (Hopper, closest supported), 9.0 -> use "90"
#
# Build time: ~30-60 min with single arch vs 2-4 hours with all archs
ARG CUDA_ARCH=90
RUN echo "Building for CUDA architecture: sm_${CUDA_ARCH}"
RUN cmake .. \
-GNinja \
-DCMAKE_BUILD_TYPE=Release \
@@ -107,7 +115,9 @@ RUN cmake .. \
-DON_INFER=OFF \
-DWITH_PYTHON=ON \
-DWITH_AVX=OFF \
-DCUDA_ARCH_NAME=Auto \
-DCUDA_ARCH_NAME=Manual \
-DCUDA_ARCH_BIN="${CUDA_ARCH}" \
-DCMAKE_CUDA_ARCHITECTURES="${CUDA_ARCH}" \
-DCMAKE_EXPORT_COMPILE_COMMANDS=ON
# Build PaddlePaddle (this takes 2-4 hours)