File tree Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Expand file tree Collapse file tree 1 file changed +8
-1
lines changed Original file line number Diff line number Diff line change @@ -468,8 +468,15 @@ ENV UV_LINK_MODE=copy
468468
469469RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \
470470 && echo 'tzdata tzdata/Zones/America select Los_Angeles' | debconf-set-selections \
471+ && CUDA_VERSION_DASH=$(echo $CUDA_VERSION | cut -d. -f1,2 | tr '.' '-' ) \
471472 && apt-get update -y \
472- && apt-get install -y git
473+ && apt-get install -y git \
474+ cuda-nvrtc-dev-${CUDA_VERSION_DASH} \
475+ libcublas-dev-${CUDA_VERSION_DASH} \
476+ libcurand-dev-${CUDA_VERSION_DASH} \
477+ libcusolver-dev-${CUDA_VERSION_DASH} \
478+ libcusparse-dev-${CUDA_VERSION_DASH} && \
479+ rm -rf /var/lib/apt/lists/*
473480
474481# install development dependencies (for testing)
475482RUN --mount=type=cache,target=/root/.cache/uv \
You can’t perform that action at this time.
0 commit comments