vedithal-amd
a926660670
* Use TheRock nightly builds in testing container * Add HIP_DEVICE_LIB_PATH env var for hipcc to work * Add HIP_PLATFORM env var for cmake hip package * Add tarball placeholder * Add -f to curl command to fail on HTTP error
53 wiersze
1.9 KiB
Docker
53 wiersze
1.9 KiB
Docker
# Use a base image
|
|
FROM ubuntu:22.04
|
|
|
|
# Install curl first (needed for ROCm download)
|
|
RUN apt-get update && apt-get install -y curl
|
|
|
|
# Define the tarball name as a variable
|
|
# Check https://therock-nightly-tarball.s3.amazonaws.com/index.html for latest builds
|
|
# Use therock-dist-linux-gfx<arch>-dcgpu-<rocm-version>.tar.gz naming convention
|
|
ARG TARBALL_NAME=<tarball_name>
|
|
|
|
# Install ROCm from TheRock Nightly build
|
|
RUN mkdir -p /rocm && \
|
|
curl -fLO https://therock-nightly-tarball.s3.amazonaws.com/${TARBALL_NAME} && \
|
|
tar -xf ${TARBALL_NAME} -C /rocm && \
|
|
rm ${TARBALL_NAME}
|
|
|
|
# Set environment variables for ROCm
|
|
ENV PATH="/rocm/bin:${PATH}" \
|
|
ROCM_PATH="/rocm" \
|
|
LD_LIBRARY_PATH="/rocm/lib:/rocm/lib/rocm_sysdeps/lib:${LD_LIBRARY_PATH}" \
|
|
HIP_DEVICE_LIB_PATH="/rocm/llvm/amdgcn/bitcode" \
|
|
HIP_PLATFORM=amd
|
|
|
|
# Update package list and install prerequisites
|
|
RUN apt-get update && apt-get install -y \
|
|
software-properties-common cmake locales git \
|
|
&& add-apt-repository ppa:deadsnakes/ppa \
|
|
&& apt-get update
|
|
|
|
# Generate the desired locale
|
|
RUN locale-gen en_US.UTF-8
|
|
|
|
# Install Python 3.10 and pip
|
|
RUN apt-get install -y python3.10 python3.10-venv python3.10-dev python3-pip libsqlite3-dev
|
|
RUN python3.10 -m venv /venv
|
|
ENV PATH="/venv/bin:$PATH"
|
|
RUN python -m pip install --upgrade pip
|
|
|
|
# Install any rocprofiler-compute dependencies specified in requirements.txt
|
|
COPY projects/rocprofiler-compute/requirements.txt /app/projects/rocprofiler-compute/requirements.txt
|
|
COPY projects/rocprofiler-compute/requirements-test.txt /app/projects/rocprofiler-compute/requirements-test.txt
|
|
RUN python -m pip install -r /app/projects/rocprofiler-compute/requirements.txt -r /app/projects/rocprofiler-compute/requirements-test.txt
|
|
|
|
# Set the working directory
|
|
WORKDIR /app
|
|
|
|
# Allows running git commands in /app
|
|
RUN git config --global --add safe.directory /app
|
|
|
|
# Run interactive bash shell
|
|
CMD ["/bin/bash"]
|