mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-17 21:10:43 +00:00
70 lines
2.9 KiB
Text
70 lines
2.9 KiB
Text
FROM quay.io/pypa/manylinux2010_x86_64:2020-04-04-f427f46
|
|
|
|
ARG PYTHON_VERSION=3.5
|
|
|
|
ENV NVIDIA_VISIBLE_DEVICES all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
|
|
|
|
ADD scripts /tmp/scripts
|
|
RUN cd /tmp/scripts && /tmp/scripts/install_centos.sh && /tmp/scripts/install_deps.sh -p $PYTHON_VERSION && \
|
|
yum remove -y devtoolset-\* && yum install -y devtoolset-7-binutils devtoolset-7-gcc devtoolset-7-gcc-c++ devtoolset-7-gcc-gfortran devtoolset-7-libquadmath-devel devtoolset-7-libstdc++-devel devtoolset-7-runtime && \
|
|
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.1.243
|
|
ENV CUDA_PKG_VERSION 10-1-$CUDA_VERSION-1
|
|
RUN yum install -y \
|
|
cuda-cudart-$CUDA_PKG_VERSION \
|
|
cuda-compat-10-1 \
|
|
cuda-libraries-$CUDA_PKG_VERSION \
|
|
cuda-nvtx-$CUDA_PKG_VERSION \
|
|
libcublas10-10.2.1.243-1 \
|
|
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 \
|
|
libcublas-devel-10.2.1.243-1 \
|
|
&& \
|
|
ln -s cuda-10.1 /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:/opt/rh/devtoolset-7/root/usr/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.1 brand=tesla,driver>=384,driver<385 brand=tesla,driver>=396,driver<397 brand=tesla,driver>=410,driver<411"
|
|
|
|
|
|
ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs
|
|
|
|
ENV CUDNN_VERSION 7.6.5.32
|
|
LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}"
|
|
|
|
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
|
|
RUN CUDNN_DOWNLOAD_SUM=7eaec8039a2c30ab0bc758d303588767693def6bf49b22485a2c00bf2e136cb3 && \
|
|
curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v7.6.5/cudnn-10.1-linux-x64-v7.6.5.32.tgz -O && \
|
|
echo "$CUDNN_DOWNLOAD_SUM cudnn-10.1-linux-x64-v7.6.5.32.tgz" | sha256sum -c - && \
|
|
tar --no-same-owner -xzf cudnn-10.1-linux-x64-v7.6.5.32.tgz -C /usr/local && \
|
|
rm cudnn-10.1-linux-x64-v7.6.5.32.tgz && \
|
|
ldconfig
|
|
|
|
|
|
ARG BUILD_UID=1000
|
|
ARG BUILD_USER=onnxruntimedev
|
|
RUN adduser --comment 'onnxruntime Build User' --uid $BUILD_UID $BUILD_USER
|
|
WORKDIR /home/$BUILD_USER
|
|
USER $BUILD_USER
|