stable-baselines3/scripts/run_docker_gpu.sh

#!/bin/bash
# Launch an experiment using the docker gpu image
cmd_line="$@"
echo "Executing in the docker (gpu image):"
echo $cmd_line

# Using new-style GPU argument
NVIDIA_ARG="--gpus all"

docker run -it ${NVIDIA_ARG} --rm --network host --ipc=host \
--mount src=$(pwd),target=/home/mamba/stable-baselines3,type=bind stablebaselines/stable-baselines3:latest \
bash -c "cd /home/mamba/stable-baselines3/ && $cmd_line"
Add dockerfile + update doc 2020-05-07 09:08:49 +00:00			`#!/bin/bash`
			`# Launch an experiment using the docker gpu image`
			`cmd_line="$@"`
			`echo "Executing in the docker (gpu image):"`
			`echo $cmd_line`

Simplify GPU detection in docker launch script (#2064) Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org> 2025-01-07 13:58:28 +00:00			`# Using new-style GPU argument`
			`NVIDIA_ARG="--gpus all"`
Add dockerfile + update doc 2020-05-07 09:08:49 +00:00
			`docker run -it ${NVIDIA_ARG} --rm --network host --ipc=host \`
Simplify GPU detection in docker launch script (#2064) Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org> 2025-01-07 13:58:28 +00:00			`--mount src=$(pwd),target=/home/mamba/stable-baselines3,type=bind stablebaselines/stable-baselines3:latest \`
			`bash -c "cd /home/mamba/stable-baselines3/ && $cmd_line"`