mirror of
https://github.com/ROCm/ROCm-docker.git
synced 2025-12-24 14:21:42 +01:00
4f2241c1a8
* Adding almalinux-8 dockerfile * Install multi-install versioned ROCm * Bug * Install python (needed by Magma install) and python3-dev (needed to be able to run PyTorch in compile mode when testing PyTorch wheels in resulting docker image) * yum cannot find python3-dev on almalinux * Use ln since no 'python' package exists * Move python link creation to ensure it persists * Don't install python link here; we will do it in manylinux image since it's only really needed for Magma build currently * comment * Add almalinux image creation to build_all.sh * Push almalinux8 image * Add Devtoolset 11 paths (cherry picked from commit 5126740ec09bf0b86d0c4edace90dd317ca9e244) --------- Co-authored-by: Pruthvi Madugundu <pruthvigithub@gmail.com>
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}
|