stable-baselines3/scripts/run_docker_cpu.sh

12 lines
357 B
Bash
Raw Normal View History

2020-05-07 09:08:49 +00:00
#!/bin/bash
# Launch an experiment using the docker cpu image
cmd_line="$@"
echo "Executing in the docker (cpu image):"
echo $cmd_line
docker run -it --rm --network host --ipc=host \
--mount src=$(pwd),target=/root/code/stable-baselines3,type=bind stablebaselines/stable-baselines3-cpu:latest \
bash -c "cd /root/code/stable-baselines3/ && $cmd_line"