diff --git a/start_interactively.sh b/start_interactively.sh index 0618671..d8580fa 100644 --- a/start_interactively.sh +++ b/start_interactively.sh @@ -1,3 +1,3 @@ #!/bin/bash -docker run -it --rm --gpus all -p 8900:8888 -v /home/${USER}:/wd ${USER}_${IMAGE_NAME} +docker run -it --rm --gpus "device=0" -p 8900:8888 -v /home/${USER}:/wd ${USER}_${IMAGE_NAME} \ No newline at end of file