ARG OS_VERSION=16.04
FROM ubuntu:${OS_VERSION}
ARG ROCM_VERSION=3.5.1
RUN <<EOF
apt-get update
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates curl gnupg
curl -sL http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -
echo "deb [arch=amd64] http://repo.radeon.com/rocm/apt/${ROCM_VERSION}/ xenial main" > /etc/apt/sources.list.d/rocm.list
apt-get update
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends sudo libelf1 libnuma-dev build-essential git vim-nox cmake-curses-gui kmod file python3 python3-pip rocm-dev
apt-get clean
rm -rf /var/lib/apt/lists/*
EOF
COPY sudo-nopasswd /etc/sudoers.d/sudo-nopasswd
COPY path-fix /etc/sudoers.d/path-fix
RUN useradd --create-home -G sudo,video --shell /bin/bash rocm
USER rocm
WORKDIR /home/rocm
ENV PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/opt/rocm/bin
CMD ["/bin/bash", "-l"]