amd_python_installer/Dockerfile.build at main · gbmerrall/amd_python_installer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
ARG DOCKER_ROCM_VERSION=6.4.3
FROM rocm/dev-ubuntu-24.04:${DOCKER_ROCM_VERSION}

ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONUNBUFFERED=1
ENV AMD_LOG_LEVEL=0
ENV ROCM_PATH=/opt/rocm
ENV PATH=$PATH:$ROCM_PATH/bin:$ROCM_PATH/hip/bin
ENV LD_LIBRARY_PATH=$ROCM_PATH/lib:$ROCM_PATH/hip/lib:$ROCM_PATH/opencl/lib
ENV HIP_VISIBLE_DEVICES=0,1
ENV CUDA_VISIBLE_DEVICES=0,1
ENV PYTORCH_ROCM_DEVICE=0,1
ENV HSA_OVERRIDE_GFX_VERSION=11.0.0
ENV HSA_ENABLE_SDMA=0
ENV GPU_MAX_HEAP_SIZE=100
ENV GPU_MAX_ALLOC_PERCENT=100
ENV ROCM_HOME=$ROCM_PATH
ENV CUDA_HOME=$ROCM_PATH
ENV HSA_TOOLS_LIB=librocprofiler64.so

RUN curl -fsSL https://bazel.build/bazel-release.pub.gpg | gpg --dearmor > /usr/share/keyrings/bazel-archive-keyring.gpg
RUN echo "deb [arch=amd64 signed-by=/usr/share/keyrings/bazel-archive-keyring.gpg] https://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list

# Install build tools for Python and C++
RUN apt-get update && apt-get install -y \
    build-essential \
    python3 \
    python3-dev \
    python3-pip \
    python3-venv \
    cmake \
    pkg-config \
    git \
    libprotobuf-dev \
    ninja-build \
    libopenmpi-dev \
    openmpi-bin \
    libnuma-dev \
    pciutils \
    mesa-utils \
    clinfo \
    lsb-release \
    gnupg \
    g++ \
    curl \
    gnupg \
    software-properties-common \
    apt-transport-https \
    ca-certificates \
    pybind11-dev \
    libprotobuf-dev \
    protobuf-compiler \
    jq \
    hipblas-dev \
    rocblas-dev \
    miopen-hip-dev \
    hiprand-dev \
    hipfft-dev \
    hipsolver-dev \
    rccl-dev \
    migraphx-dev \
    hipsparse-dev \
    hipsparselt-dev \
    hipcub-dev \
    rocthrust-dev \
    ccache \
    ffmpeg \
    libpng-dev \
    libjpeg-dev \
    libwebp-dev \
    libnvjpeg12 \
    libswresample-dev \
    libavcodec-dev \
    libavutil-dev \
    libavformat-dev \
    libswscale-dev \
    python-is-python3 \
    bazel \
    clang-18 \
    llvm-18 \
    libmlir-18-dev \
    mlir-18-tools \
    lld \
    && apt clean \
    && rm -rf /var/lib/apt/lists/*

# Create directory for wheel output
RUN mkdir -p /wheels

# Set working directory
WORKDIR /build


# Volume for saving wheels to local filesystem
VOLUME ["/wheels"]

# Default command
CMD ["bash"]