mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
179603775f
Use CUDA 10.1 for Linux build (Windows change is already in) Please note, cublas 10.2.1.243 is for CUDA SDK 10.1.243, not CUDA 10.2.x. CUDA 10.2.89 need cublas 10.2.2.89. They match on the last part of the digits. libcublas10-10.1.0.105 won't work!!! The cuda docker image by viswamy is already using 10.1, no need to change.
67 lines
2.7 KiB
Text
67 lines
2.7 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.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:${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 /usr/sbin/adduser --uid $BUILD_UID $BUILD_USER
|
|
WORKDIR /home/$BUILD_USER
|
|
USER $BUILD_USER
|