mirror of
https://github.com/saymrwulf/stable-baselines3.git
synced 2026-05-14 20:58:03 +00:00
e071796549
* sets replay buffer device after loading * update changelog * update changelog * correct changelog * add test for replay buffer device * Fix test to actually test the bug fix * [ci skip] Update version * [ci skip] Update docker images --------- Co-authored-by: PatrickHelm <patrick.helm@gmx.net> Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
25 lines
738 B
Bash
Executable file
25 lines
738 B
Bash
Executable file
#!/bin/bash
|
|
|
|
CPU_PARENT=mambaorg/micromamba:1.5-jammy
|
|
GPU_PARENT=mambaorg/micromamba:1.5-jammy-cuda-11.7.1
|
|
|
|
TAG=stablebaselines/stable-baselines3
|
|
VERSION=$(cat ./stable_baselines3/version.txt)
|
|
|
|
if [[ ${USE_GPU} == "True" ]]; then
|
|
PARENT=${GPU_PARENT}
|
|
PYTORCH_DEPS="pytorch-cuda=11.7"
|
|
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
|