feat(nasp): fix nvidia for docker
This commit is contained in:
@@ -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:"
|
||||
|
||||
Reference in New Issue
Block a user