diff --git a/start_daemon.sh b/start_daemon.sh index 713364e..6e8d780 100755 --- a/start_daemon.sh +++ b/start_daemon.sh @@ -1,4 +1,4 @@ #!/bin/bash CURDIRNAME=${PWD##*/} -docker run --name ${USER}_${CURDIRNAME} -d --gpus all -p 8888:8888 -v $(pwd):/wd --name ${USER}_${CURDIRNAME} ${USER}_${CURDIRNAME} \ No newline at end of file +docker run -d --gpus "device=0" -p 8888:8888 -v $(pwd):/wd --name ${USER}_${CURDIRNAME} ${USER}_${CURDIRNAME} \ No newline at end of file