code_public/lidar_rendu
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update docs for GPU selection, fix test assertions + test_pipeline

Browse Source
This commit is contained in:
Antoine Jacquin
2026-05-31 17:19:53 +02:00
parent eb9545b56d
commit 07d2a8aea9
3 changed files with 21 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
Dockerfile
View File

@ -46,6 +46,11 @@ RUN pip3 install --no-cache-dir \
cmcrameri
# Install CuPy for GPU acceleration (optional - will fallback to numpy if not available)
# CUPY_CUDA_COMPILE_WITH_CACHE=1 enables JIT compilation of kernels for
# GPU architectures not included in the pre-built wheel (e.g. sm_89, sm_120).
# The devel image includes nvcc for JIT compilation. If CuPy still fails
# (CUDA_ERROR_NO_BINARY_FOR_GPU), the pipeline falls back to CPU automatically.
ENV CUPY_CUDA_COMPILE_WITH_CACHE=1
RUN pip3 install --no-cache-dir cupy-cuda12x || echo "CuPy not available - GPU acceleration disabled"
# Copy and install the pipeline package