improve docker scripts (#23)
* Feature/support x86 64 (#22) * [Add] support for x86_64 archtecture. * make script executable * [fix] apply shellcheck suggestions (to get rid of warning during runtime) * [fix] update the image_tag to match the x86 tag * add input detection as optional, rename scripts --------- Co-authored-by: Austin Gregg-Smith <blooop@gmail.com>
This commit is contained in:
committed by
GitHub
parent
ac02893dee
commit
21d434f608
65
docker/start_user_docker.sh
Executable file
65
docker/start_user_docker.sh
Executable file
@@ -0,0 +1,65 @@
|
||||
#!/bin/bash
|
||||
##
|
||||
## Copyright (c) 2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||||
##
|
||||
## NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
|
||||
## property and proprietary rights in and to this material, related
|
||||
## documentation and any modifications thereto. Any use, reproduction,
|
||||
## disclosure or distribution of this material and related documentation
|
||||
## without an express license agreement from NVIDIA CORPORATION or
|
||||
## its affiliates is strictly prohibited.
|
||||
##
|
||||
|
||||
|
||||
input_arg="$1"
|
||||
|
||||
|
||||
if [ -z "$input_arg" ]; then
|
||||
echo "Argument empty, trying to run based on architecture"
|
||||
arch=$(uname -m)
|
||||
if [ "$arch" == "x86_64" ]; then
|
||||
input_arg="x86"
|
||||
elif [ "$arch" == "arm64" ]; then
|
||||
input_arg="aarch64"
|
||||
elif [ "$arch" == "aarch64" ]; then
|
||||
input_arg="aarch64"
|
||||
fi
|
||||
fi
|
||||
|
||||
|
||||
if [ "$input_arg" == "x86" ]; then
|
||||
|
||||
docker run --rm -it \
|
||||
--privileged \
|
||||
-e NVIDIA_DISABLE_REQUIRE=1 \
|
||||
-e NVIDIA_DRIVER_CAPABILITIES=all --device /dev/dri \
|
||||
--mount type=bind,src=/home/$USER/code,target=/home/$USER/code \
|
||||
--hostname ros1-docker \
|
||||
--add-host ros1-docker:127.0.0.1 \
|
||||
--gpus all \
|
||||
--network host \
|
||||
--env DISPLAY=unix$DISPLAY \
|
||||
--volume /tmp/.X11-unix:/tmp/.X11-unix \
|
||||
--volume /dev:/dev \
|
||||
curobo_docker:user_$1
|
||||
|
||||
elif [ "$input_arg" == "aarch64" ]; then
|
||||
|
||||
docker run --rm -it \
|
||||
--runtime nvidia \
|
||||
--hostname ros1-docker \
|
||||
--add-host ros1-docker:127.0.0.1 \
|
||||
--network host \
|
||||
--gpus all \
|
||||
--env ROS_HOSTNAME=localhost \
|
||||
--env DISPLAY=$DISPLAY \
|
||||
--volume /tmp/.X11-unix:/tmp/.X11-unix \
|
||||
--volume /dev/input:/dev/input \
|
||||
--mount type=bind,src=/home/$USER/code,target=/home/$USER/code \
|
||||
curobo_docker:user_$1
|
||||
|
||||
elif [[ "$input_arg" == *isaac_sim* ]] ; then
|
||||
echo "Isaac Sim User Docker is not supported"
|
||||
else
|
||||
echo "Unknown docker"
|
||||
fi
|
||||
Reference in New Issue
Block a user