Verified Commit 1725cf09 authored by Jean-Noël Grad's avatar Jean-Noël Grad
Browse files

Pin ROCm repository

This guarantees apt fetches all ROCm packages from the same release.
parent f9d0a069
......@@ -2,7 +2,10 @@ ARG BASE_VERSION=3.3
FROM rocm/dev-ubuntu-18.04:${BASE_VERSION}
ENV DEBIAN_FRONTEND noninteractive
COPY ubuntu-packages.txt /tmp
RUN apt-get update && xargs -a /tmp/ubuntu-packages.txt apt-get install --no-install-recommends -y \
RUN apt_folder=$([ "${BASE_VERSION}" = "latest" ] && echo "debian" || echo "${BASE_VERSION}") \
&& sed -i "s||${apt_folder}/|" /etc/apt/sources.list.d/rocm.list \
&& apt-get update && xargs -a /tmp/ubuntu-packages.txt apt-get install --no-install-recommends -y \
&& apt-get install --no-install-recommends -y \
rocrand rocfft rocthrust rocblas rocsolver kmod \
&& apt-get clean \
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment