34 lines
1.2 KiB
Bash
Executable File
34 lines
1.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
cd "$(dirname "$0")"
|
|
|
|
VERSION=$(date +%Y%m%d)-latest
|
|
echo Building: ${VERSION}
|
|
|
|
|
|
###### Training Image ######
|
|
# docker build -t dcloud/lerobot-train:${VERSION} -f docker/Dockerfile.train . --build-arg http_proxy=http://192.168.16.68:18000 --build-arg https_proxy=http://192.168.16.68:18000
|
|
# docker run -it --rm --gpus '"device=7"' \
|
|
# -v ${PWD}/docker/inputs:/workspace/inputs \
|
|
# -v ${PWD}/docker/outputs:/workspace/outputs/checkpoints \
|
|
# -v ${PWD}/docker/train_task.json:/workspace/inputs/task.json \
|
|
# --shm-size=128G \
|
|
# dcloud/lerobot-train:${VERSION}
|
|
|
|
|
|
###### Merge Image ######
|
|
docker build -t dcloud/lerobot-merge:${VERSION} -f docker/Dockerfile.merge . --build-arg http_proxy=http://192.168.16.68:18000 --build-arg https_proxy=http://192.168.16.68:18000
|
|
docker run -it --rm \
|
|
-v ${PWD}/docker/inputs:/workspace/inputs \
|
|
-v ${PWD}/docker/outputs:/workspace/outputs \
|
|
-v ${PWD}/docker/merge_task.json:/workspace/inputs/task.json \
|
|
--shm-size=128G \
|
|
dcloud/lerobot-merge:${VERSION}
|
|
|
|
|
|
# # Remove dangling images
|
|
docker rmi $(docker images -f "dangling=true" -q)
|
|
docker images | grep lerobot | grep -v ${VERSION} | awk '{print $1":"$2}' | xargs docker rmi
|