mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
66 lines
2.6 KiB
Text
66 lines
2.6 KiB
Text
FROM mcr.microsoft.com/dotnet-buildtools/prereqs:centos-6-50f0d02-20190918213956
|
|
|
|
ENV NVIDIA_VISIBLE_DEVICES all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
|
|
|
|
ADD scripts /tmp/scripts
|
|
RUN cd /tmp/scripts && /tmp/scripts/install_centos.sh
|
|
ENV PATH /opt/rh/devtoolset-2/root/usr/bin:${PATH}
|
|
RUN /tmp/scripts/install_deps.sh -p 3.6 && rm -rf /tmp/scripts
|
|
|
|
#Below are copied from https://gitlab.com/nvidia/container-images/cuda/tree/master/dist/centos6
|
|
|
|
RUN NVIDIA_GPGKEY_SUM=d1be581509378368edeec8c1eb2958702feedf3bc3d17011adbf24efacce4ab5 && \
|
|
curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/rhel6/x86_64/7fa2af80.pub | sed '/^Version/d' > /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA && \
|
|
echo "$NVIDIA_GPGKEY_SUM /etc/pki/rpm-gpg/RPM-GPG-KEY-NVIDIA" | sha256sum -c -
|
|
|
|
COPY cuda_manylinux2010.repo /etc/yum.repos.d/cuda.repo
|
|
|
|
ENV CUDA_VERSION 10.0.130
|
|
|
|
ENV CUDA_PKG_VERSION 10-0-$CUDA_VERSION-1
|
|
RUN yum install -y \
|
|
cuda-cudart-$CUDA_PKG_VERSION \
|
|
cuda-libraries-$CUDA_PKG_VERSION \
|
|
cuda-nvtx-$CUDA_PKG_VERSION \
|
|
cuda-nvml-dev-$CUDA_PKG_VERSION \
|
|
cuda-command-line-tools-$CUDA_PKG_VERSION \
|
|
cuda-libraries-dev-$CUDA_PKG_VERSION \
|
|
cuda-minimal-build-$CUDA_PKG_VERSION \
|
|
&& \
|
|
ln -s cuda-10.0 /usr/local/cuda && \
|
|
rpm -e --nodeps gcc gcc-c++ && \
|
|
rm -rf /var/cache/yum/*
|
|
|
|
# nvidia-docker 1.0
|
|
RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && \
|
|
echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
|
|
|
|
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
|
|
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}
|
|
|
|
# nvidia-container-runtime
|
|
ENV NVIDIA_VISIBLE_DEVICES all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
|
|
ENV NVIDIA_REQUIRE_CUDA "cuda>=10.0 brand=tesla,driver>=384,driver<385 brand=tesla,driver>=410,driver<411"
|
|
|
|
|
|
ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
|
|
|
|
ENV CUDNN_VERSION 7.6.4.38
|
|
LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}"
|
|
|
|
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
|
|
RUN CUDNN_DOWNLOAD_SUM=417bb5daf51377037eb2f5c87649000ca1b9cec0acb16cfe07cb1d3e9a961dbf && \
|
|
curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v7.6.4/cudnn-10.0-linux-x64-v7.6.4.38.tgz -O && \
|
|
echo "$CUDNN_DOWNLOAD_SUM cudnn-10.0-linux-x64-v7.6.4.38.tgz" | sha256sum -c - && \
|
|
tar --no-same-owner -xzf cudnn-10.0-linux-x64-v7.6.4.38.tgz -C /usr/local && \
|
|
rm cudnn-10.0-linux-x64-v7.6.4.38.tgz && \
|
|
/sbin/ldconfig
|
|
|
|
|
|
ARG BUILD_UID=1000
|
|
ARG BUILD_USER=onnxruntimedev
|
|
RUN /usr/sbin/adduser --uid $BUILD_UID $BUILD_USER
|
|
WORKDIR /home/$BUILD_USER
|
|
USER $BUILD_USER
|