21 lines
913 B
Docker
21 lines
913 B
Docker
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"]
|