mirror of
https://github.com/zama-ai/concrete.git
synced 2026-01-08 04:23:57 -05:00
22 lines
720 B
Docker
22 lines
720 B
Docker
FROM quay.io/pypa/manylinux_2_28_x86_64:2024-02-08-a1b4ddc AS build
|
|
|
|
ARG PACKAGEURL=https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-rhel8-11-8-local-11.8.0_520.61.05-1.x86_64.rpm
|
|
|
|
RUN dnf install -y kernel-devel kernel-headers && dnf clean all
|
|
RUN curl ${PACKAGEURL} -o cuda.rpm
|
|
RUN rpm -i cuda.rpm \
|
|
&& dnf install -y epel-release \
|
|
&& dnf clean all
|
|
# hadolint ignore=DL3041
|
|
RUN dnf update -y \
|
|
&& dnf -y module install nvidia-driver:latest-dkms \
|
|
&& dnf -y install cuda \
|
|
&& dnf clean all
|
|
|
|
FROM scratch
|
|
|
|
ARG CUDAVERSION=11.8
|
|
|
|
COPY --from=build /usr/local/cuda-${CUDAVERSION}/ /usr/local/cuda-${CUDAVERSION}/
|
|
COPY --from=build /usr/lib64/libcuda.so* /usr/lib64/
|