feat(nasp): fix nvidia for docker

This commit is contained in:
2024-06-12 11:03:28 +08:00
parent 3d70d0cec5
commit a40aaa9233
2 changed files with 12 additions and 15 deletions

View File

@@ -44,7 +44,7 @@ fi
ret=$(ask_for_yN "Use GPU?")
if [[ "$ret" == "1" ]]; then
base_command="$base_command \\
--runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all"
--device nvidia.com/gpu=all"
fi
ret=$(ask_for_yN "Use RDMA?")
if [[ "$ret" == "1" ]]; then
@@ -52,21 +52,11 @@ if [[ "$ret" == "1" ]]; then
--device=/dev/infiniband/uverbs0"
fi
read -p "Image name (default: git.nasp.fit/nasp/nasp-ubuntu): "
if [[ -z "$REPLY" ]]; then
image_name="git.nasp.fit/nasp/nasp-ubuntu"
else
image_name="$REPLY"
fi
image_name="git.nasp.fit/nasp/nasp-ubuntu"
base_command="$base_command \\
-it $image_name"
read -p "Start up command (default: /etc/startup.sh): "
if [[ -z "$REPLY" ]]; then
cmd="/etc/startup.sh"
else
cmd="$REPLY"
fi
cmd="/etc/startup.sh"
base_command="$base_command $cmd"
echo "Will run:"