mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-05-14 20:58:03 +00:00
25 lines
722 B
Bash
Executable file
25 lines
722 B
Bash
Executable file
#!/bin/bash
|
|
|
|
CPU_PARENT=ubuntu:16.04
|
|
GPU_PARENT=nvidia/cuda:10.1-cudnn7-runtime-ubuntu16.04
|
|
|
|
TAG=stablebaselines/stable-baselines3
|
|
VERSION=$(cat ./stable_baselines3/version.txt)
|
|
|
|
if [[ ${USE_GPU} == "True" ]]; then
|
|
PARENT=${GPU_PARENT}
|
|
PYTORCH_DEPS="cudatoolkit=10.1"
|
|
else
|
|
PARENT=${CPU_PARENT}
|
|
PYTORCH_DEPS="cpuonly"
|
|
TAG="${TAG}-cpu"
|
|
fi
|
|
|
|
echo "docker build --build-arg PARENT_IMAGE=${PARENT} --build-arg PYTORCH_DEPS=${PYTORCH_DEPS} -t ${TAG}:${VERSION} ."
|
|
docker build --build-arg PARENT_IMAGE=${PARENT} --build-arg PYTORCH_DEPS=${PYTORCH_DEPS} -t ${TAG}:${VERSION} .
|
|
docker tag ${TAG}:${VERSION} ${TAG}:latest
|
|
|
|
if [[ ${RELEASE} == "True" ]]; then
|
|
docker push ${TAG}:${VERSION}
|
|
docker push ${TAG}:latest
|
|
fi
|