Fix docker GPU (#492)

* Update docker command

* Add default arg to dockerfile

Author: araffin

docker/Dockerfile

@@ -1,4 +1,4 @@
-ARG PARENT_IMAGE
+ARG PARENT_IMAGE=stablebaselines/stable-baselines3
 FROM $PARENT_IMAGE
 
 USER root

rl_zoo3/version.txt

@@ -1 +1 @@
-2.6.0
+2.6.1a0

scripts/run_docker_gpu.sh

@@ -8,6 +8,6 @@ echo $cmd_line
 
 # Note: --user=root is needed, as the current user id/group id will be used
 # to mount the log directory (and $MAMBAUSER is not root)
-docker run -it --user=root --runtime=nvidia --rm --network host --ipc=host \
+docker run -it --user=root --gpus all --rm --network host --ipc=host \
   --mount src=$(pwd),target=/home/mambauser/code/rl_zoo3,type=bind stablebaselines/rl-baselines3-zoo:latest\
   bash -c "cd /home/mambauser/code/rl_zoo3/ && $cmd_line"