Update Ubuntu 18.04 GPU container

This commit is contained in:
Richard Berger 2020-06-30 09:38:28 -04:00
parent a73f1d4f03
commit af2e1bf7c6
No known key found for this signature in database
GPG Key ID: A9E83994E0BA0CAB
1 changed files with 19 additions and 4 deletions

View File

@ -1,5 +1,5 @@
BootStrap: docker
From: rocm/dev-ubuntu-18.04
From: ubuntu:18.04
%environment
export PATH=/usr/lib/ccache:/usr/local/cuda-10.2/bin:${PATH}:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64
@ -11,6 +11,22 @@ From: rocm/dev-ubuntu-18.04
export DEBIAN_FRONTEND=noninteractive
apt-get update
apt-get upgrade --no-install-recommends -y
apt-get install -y --no-install-recommends curl libnuma-dev gnupg
curl -sL http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | apt-key add -
printf "deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main" > /etc/apt/sources.list.d/rocm.list
apt-get update
apt-get install --no-install-recommends -y \
kmod \
file \
sudo \
libelf1 \
rocm-dev \
rocm-libs \
build-essential
apt-get install --no-install-recommends -y software-properties-common
apt-get install --no-install-recommends -y \
bc \
@ -57,7 +73,6 @@ From: rocm/dev-ubuntu-18.04
python3-pkg-resources \
python3-setuptools \
python3-virtualenv \
rocm-libs \
rsync \
ssh \
vim-nox \
@ -86,10 +101,10 @@ From: rocm/dev-ubuntu-18.04
libcublas-dev
export PATH=$PATH:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64
git clone -b master-rocm-3.3 https://github.com/ROCmSoftwarePlatform/hipCUB.git
git clone -b master-rocm-3.5 https://github.com/ROCmSoftwarePlatform/hipCUB.git
mkdir hipCUB/build
cd hipCUB/build
CXX=hcc cmake -D BUILD_TEST=off ..
CXX=hipcc cmake -D BUILD_TEST=off ..
make -j4
make package
make install