feat: add host c1
This commit is contained in:
@@ -42,17 +42,20 @@ if [[ "$EUID" == "0" ]]; then
|
||||
exit 1
|
||||
fi
|
||||
|
||||
ret=$(ask_for_yN "Use GPU?")
|
||||
if [[ "$ret" == "1" ]]; then
|
||||
base_command="$base_command \\
|
||||
--device nvidia.com/gpu=all"
|
||||
if command -v nvidia-smi 1>/dev/null; then
|
||||
ret=$(ask_for_yN "Use GPU?")
|
||||
if [[ "$ret" == "1" ]]; then
|
||||
base_command="$base_command \\
|
||||
--device nvidia.com/gpu=all"
|
||||
fi
|
||||
fi
|
||||
ret=$(ask_for_yN "Use RDMA?")
|
||||
if [[ "$ret" == "1" ]]; then
|
||||
base_command="$base_command \\
|
||||
--device=/dev/infiniband/uverbs0"
|
||||
if command -v ibv_devinfo 1>/dev/null; then
|
||||
ret=$(ask_for_yN "Use RDMA?")
|
||||
if [[ "$ret" == "1" ]]; then
|
||||
base_command="$base_command \\
|
||||
--device=/dev/infiniband/uverbs0"
|
||||
fi
|
||||
fi
|
||||
|
||||
image_name="git.nasp.fit/nasp/nasp-ubuntu"
|
||||
base_command="$base_command \\
|
||||
-it $image_name"
|
||||
|
||||
Reference in New Issue
Block a user