Update launch_container.sh

linjieli222 · Sep 16, 2021 · 32c1c52 · 32c1c52
1 parent b220d37
commit 32c1c52
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/launch_container.sh b/launch_container.sh
@@ -19,7 +19,6 @@ else
 fi
 
 docker run --gpus '"'device=$CUDA_VISIBLE_DEVICES'"' --ipc=host --network=host --rm -it \
-    --device /dev/nvidia0 --device /dev/nvidia-uvm --device /dev/nvidia-uvm-tools --device /dev/nvidiactl \
     --mount src=$(pwd),dst=/src,type=bind \
     --mount src=$OUTPUT,dst=/storage,type=bind \
     --mount src=$PRETRAIN_DIR,dst=/pretrain,type=bind,readonly \