47 lines
970 B
Bash
Executable File
47 lines
970 B
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )
|
|
cd $SCRIPT_DIR
|
|
|
|
. ${SCRIPT_DIR=}/../variables.sh
|
|
|
|
DOCKER_TAG=latest
|
|
|
|
# Parse arguments
|
|
while [ "$#" -gt 0 ]; do
|
|
case "$1" in
|
|
--cuda-version)
|
|
CUDA_VERSION="$2"
|
|
shift
|
|
;;
|
|
--python-version)
|
|
PYTHON_VERSION="$2"
|
|
shift
|
|
;;
|
|
--tag)
|
|
DOCKER_TAG="$2"
|
|
shift
|
|
;;
|
|
--github-repository)
|
|
if [ -n "${GITHUB_REPOSITORY}" ] && ! [[ $GITHUB_REPOSITORY =~ ^--.* ]]; then
|
|
GITHUB_REPOSITORY="$2"
|
|
shift
|
|
fi
|
|
;;
|
|
*)
|
|
echo "Unknown argument '$1'"
|
|
exit 1
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
echo "Building conda environment using python ${PYTHON_VERSION} with CUDA ${CUDA_VERSION}"
|
|
|
|
docker buildx \
|
|
build \
|
|
--progress=plain \
|
|
--build-arg CUDA_VERSION=$CUDA_VERSION \
|
|
--build-arg PYTHON_VERSION=$PYTHON_VERSION \
|
|
-t ${GITHUB_REPOSITORY}alltalk_tts_environment:${DOCKER_TAG} \
|
|
. |