mirror of
https://github.com/ROCm/ROCm-docker.git
synced 2026-01-05 04:01:12 +01:00
13 lines
240 B
Bash
13 lines
240 B
Bash
#!/bin/bash
|
|
|
|
set -ex
|
|
|
|
if [ -z $1 ]; then
|
|
echo "Need to provide ROCM_VERSION as first argument" && exit 1
|
|
fi
|
|
|
|
if [[ $1 =~ ^[0-9]+\.[0-9]+$ ]]; then
|
|
ROCM_VERSION=${1}".0"
|
|
fi
|
|
yum install -y rocm-dev${ROCM_VERSION} rocm-libs${ROCM_VERSION}
|