AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Tweaks to docker scripts [skip ci] (#28)

Browse Source
This commit is contained in:
Jonathan R. Madsen
2022-02-25 18:30:37 -06:00
committed by GitHub
parent 3151dd3aeb
commit 80e1a0d7e7
4 changed files with 20 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/build-docker.sh
+6 -2
View File
@@ -1,6 +1,6 @@
#!/usr/bin/env bash
: ${ROCM_VERSIONS:="4.5 4.3 4.3.1"}
: ${ROCM_VERSIONS:="5.0 4.5 4.3"}
: ${DISTRO:=ubuntu}
: ${VERSIONS:=20.04 18.04}
@@ -12,6 +12,10 @@ for VERSION in ${VERSIONS}
do
for i in ${ROCM_VERSIONS}
do
docker build . --tag jrmadsen/omnitrace-${DISTRO}-${VERSION}-rocm-${i} --build-arg DISTRO=${DISTRO} --build-arg VERSION=${VERSION} --build-arg ROCM_REPO_VERSION=${i}
ROCM_REPO_VERSION=${i}
if [ "${i}" = "5.0" ]; then ROCM_REPO_VERSION=debian; fi
if [ "${i}" = "4.1" ]; then ROCM_REPO_DIST="xenial"; fi
if [ "${i}" = "4.0" ]; then ROCM_REPO_DIST="xenial"; fi
docker build . --tag jrmadsen/omnitrace-${DISTRO}-${VERSION}-rocm-${i} --build-arg DISTRO=${DISTRO} --build-arg VERSION=${VERSION} --build-arg ROCM_REPO_VERSION=${ROCM_REPO_VERSION} --build-arg ROCM_REPO_DIST=${ROCM_REPO_DIST}
done
done