Replace nvidia-docker to nvidia container toolkit

Taeyoung96 · Mar 26, 2024 · 373aa88 · 373aa88
1 parent 876941c
commit 373aa88
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/docker/container_run.sh b/docker/container_run.sh
@@ -21,7 +21,8 @@ CONTAINER_NAME="$1"
 IMAGE_NAME="$2"
 
 # Launch the nvidia-docker container with the provided image name and tag
-nvidia-docker run --privileged -it \
+docker run --privileged -it \
+            --gpus all \
            -e NVIDIA_DRIVER_CAPABILITIES=all \
            -e NVIDIA_VISIBLE_DEVICES=all \
            --volume="$PROJECT_DIR:/root/ros2_ws/src" \
@@ -32,4 +33,4 @@ nvidia-docker run --privileged -it \
            --shm-size=4gb \
            --name="$CONTAINER_NAME" \
            --env="DISPLAY=$DISPLAY" \
-           "$IMAGE_NAME" /bin/bash
+           "$IMAGE_NAME" /bin/bash