mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 09:17:06 +00:00
dab19e00c1
Now build images by a script with a shared Dockerfile template --------- Co-authored-by: Binyang Li <binyli@microsoft.com> Co-authored-by: Saeed Maleki <saemal@microsoft.com>
47 lines
1.2 KiB
Bash
Executable File
47 lines
1.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -e
|
|
|
|
declare -A baseImageTable
|
|
baseImageTable=(
|
|
["cuda11.8"]="nvidia/cuda:11.8.0-devel-ubuntu20.04"
|
|
["cuda12.1"]="nvidia/cuda:12.1.1-devel-ubuntu20.04"
|
|
["cuda12.2"]="nvidia/cuda:12.2.2-devel-ubuntu20.04"
|
|
)
|
|
|
|
declare -A extraLdPathTable
|
|
extraLdPathTable=(
|
|
["cuda11.8"]="/usr/local/cuda-11.8/lib64"
|
|
["cuda12.1"]="/usr/local/cuda-12.1/compat:/usr/local/cuda-12.1/lib64"
|
|
["cuda12.2"]="/usr/local/cuda-12.2/compat:/usr/local/cuda-12.2/lib64"
|
|
)
|
|
|
|
GHCR="ghcr.io/microsoft/mscclpp/mscclpp"
|
|
TARGET=${1}
|
|
|
|
print_usage() {
|
|
echo "Usage: $0 [cuda11.8|cuda12.1|cuda12.2]"
|
|
}
|
|
|
|
if [[ ! -v "baseImageTable[${TARGET}]" ]]; then
|
|
echo "Invalid target: ${TARGET}"
|
|
print_usage
|
|
exit 1
|
|
fi
|
|
echo "Target: ${TARGET}"
|
|
|
|
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
|
|
|
|
cd ${SCRIPT_DIR}/..
|
|
|
|
docker build -t ${GHCR}:base-${TARGET} \
|
|
-f docker/base-x.dockerfile \
|
|
--build-arg BASE_IMAGE=${baseImageTable[${TARGET}]} \
|
|
--build-arg EXTRA_LD_PATH=${extraLdPathTable[${TARGET}]} \
|
|
--build-arg TARGET=${TARGET} .
|
|
|
|
docker build -t ${GHCR}:base-dev-${TARGET} \
|
|
-f docker/base-dev-x.dockerfile \
|
|
--build-arg BASE_IMAGE=${GHCR}:base-${TARGET} \
|
|
--build-arg TARGET=${TARGET} .
|