Dockerfile 3.46 KB
Newer Older
1
FROM ubuntu:20.04
Paul's avatar
Paul committed
2

Paul's avatar
Paul committed
3
ARG PREFIX=/usr/local
Paul's avatar
Paul committed
4
5
6
7

# Support multiarch
RUN dpkg --add-architecture i386

8
9
10
11
# Install rocm key
RUN apt-get update && apt-get install -y gnupg2 --no-install-recommends curl && \
    curl -sL http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add - 

Paul's avatar
Paul committed
12
# Add rocm repository
13
RUN sh -c 'echo deb [arch=amd64 trusted=yes] http://repo.radeon.com/rocm/apt/5.3/ ubuntu main > /etc/apt/sources.list.d/rocm.list'
Paul's avatar
Paul committed
14

Paul's avatar
Paul committed
15
16
17
18
# Install dependencies
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --allow-unauthenticated \
    apt-utils \
    build-essential \
bpickrel's avatar
bpickrel committed
19
    clang-format-10 \
Paul's avatar
Paul committed
20
21
22
    cmake \
    curl \
    doxygen \
Paul's avatar
Paul committed
23
    g++-7 \
Paul's avatar
Paul committed
24
    gdb \
Paul's avatar
Paul committed
25
26
    git \
    lcov \
Paul Fultz II's avatar
Paul Fultz II committed
27
28
    locales \
    pkg-config \
kahmed10's avatar
kahmed10 committed
29
30
31
    python3 \
    python3-dev \
    python3-pip \
Paul's avatar
Paul committed
32
    software-properties-common \
Paul's avatar
Paul committed
33
    wget \
Paul Fultz II's avatar
Paul Fultz II committed
34
    rocm-device-libs \
35
36
    hip-base \
    libnuma-dev \
Paul Fultz II's avatar
Paul Fultz II committed
37
38
    miopen-hip \
    rocblas \
39
40
41
42
43
44
45
46
47
48
49
50
51
52
    hipfft \
    rocthrust \
    rocrand \
    hipsparse \
    rccl \
    rccl-dev \
    rocm-smi-lib \
    rocm-dev \
    roctracer-dev \
    hipcub  \
    hipblas  \
    hipify-clang \
    half \
    libssl-dev \
Paul's avatar
Paul committed
53
    zlib1g-dev && \
Paul's avatar
Paul committed
54
55
56
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

57
58
59
# add this for roctracer dependancies
RUN pip3 install CppHeaderParser packaging==22.0

Paul Fultz II's avatar
Paul Fultz II committed
60
61
62
63
64
# Workaround broken rocm packages
RUN ln -s /opt/rocm-* /opt/rocm
RUN echo "/opt/rocm/lib" > /etc/ld.so.conf.d/rocm.conf
RUN echo "/opt/rocm/llvm/lib" > /etc/ld.so.conf.d/rocm-llvm.conf
RUN ldconfig
65

Paul Fultz II's avatar
Paul Fultz II committed
66
67
68
RUN locale-gen en_US.UTF-8
RUN update-locale LANG=en_US.UTF-8

Paul's avatar
Paul committed
69
70
71
ENV LC_ALL=C.UTF-8
ENV LANG=C.UTF-8

Shucai Xiao's avatar
Shucai Xiao committed
72
73
74
# Install dependencies
ADD dev-requirements.txt /dev-requirements.txt
ADD requirements.txt /requirements.txt
75
ADD rbuild.ini /rbuild.ini
Shucai Xiao's avatar
Shucai Xiao committed
76
77
78

COPY ./tools/install_prereqs.sh /
RUN /install_prereqs.sh /usr/local / && rm /install_prereqs.sh
79
RUN test -f /usr/local/hash || exit 1
Paul's avatar
Paul committed
80

kahmed10's avatar
kahmed10 committed
81
# Install yapf
kahmed10's avatar
kahmed10 committed
82
RUN pip3 install yapf==0.28.0
kahmed10's avatar
kahmed10 committed
83

Paul's avatar
Paul committed
84
85
86
87
# Install doc requirements
ADD doc/requirements.txt /doc-requirements.txt
RUN pip3 install -r /doc-requirements.txt

88
89
90
# Download real models to run onnx unit tests
ENV ONNX_HOME=$HOME
COPY ./tools/download_models.sh /
Shucai Xiao's avatar
Shucai Xiao committed
91
RUN /download_models.sh && rm /download_models.sh
92

Paul Fultz II's avatar
Paul Fultz II committed
93
94
# Install latest ccache version
RUN cget -p $PREFIX install facebook/zstd@v1.4.5 -X subdir -DCMAKE_DIR=build/cmake
95
RUN cget -p $PREFIX install ccache@v4.1 -DENABLE_TESTING=OFF
96
RUN cget -p /opt/cmake install kitware/cmake@v3.24.3
Paul Fultz II's avatar
Paul Fultz II committed
97

98
RUN export ONNXRT_COMMIT=$(cat test/onnx/.onnxrt-commit)
Paul Fultz II's avatar
Paul Fultz II committed
99
100

ARG ONNXRUNTIME_REPO=https://github.com/Microsoft/onnxruntime
101
ARG ONNXRUNTIME_BRANCH=main
102
103
104
105
106
ARG ONNXRUNTIME_COMMIT=$ONNXRT_COMMIT

# Let us know which commit where're using for CI
RUN echo "Onnxruntime Commit:" && echo $ONNXRUNTIME_COMMIT

Paul Fultz II's avatar
Paul Fultz II committed
107
108
109
110
111
RUN git clone --single-branch --branch ${ONNXRUNTIME_BRANCH} --recursive ${ONNXRUNTIME_REPO} onnxruntime && \
    cd onnxruntime && \
    git checkout ${ONNXRUNTIME_COMMIT} && \
    /bin/sh dockerfiles/scripts/install_common_deps.sh

112
113


Paul Fultz II's avatar
Paul Fultz II committed
114
115
ADD tools/build_and_test_onnxrt.sh /onnxruntime/build_and_test_onnxrt.sh

116
RUN cget -p /usr/local install ROCmSoftwarePlatform/rocMLIR@78b706fe9879587ab98b6614ae539265374a3fae -DBUILD_MIXR_TARGET=On -DLLVM_ENABLE_ZSTD=Off -DLLVM_ENABLE_THREADS=Off
117

118
119
ENV MIOPEN_FIND_DB_PATH=/tmp/miopen/find-db
ENV MIOPEN_USER_DB_PATH=/tmp/miopen/user-db
Paul's avatar
Paul committed
120
121
ENV LD_LIBRARY_PATH=$PREFIX/lib

Paul's avatar
Paul committed
122
123
124
# Setup ubsan environment to printstacktrace
ENV UBSAN_OPTIONS=print_stacktrace=1
ENV ASAN_OPTIONS=detect_stack_use_after_return=1:check_initialization_order=1:strict_init_order=1
Paul Fultz II's avatar
Paul Fultz II committed
125
RUN ln -s /opt/rocm/llvm/bin/llvm-symbolizer /usr/bin/llvm-symbolizer
Shucai Xiao's avatar
Shucai Xiao committed
126