42 lines
1.3 KiB
Bash
Executable File
42 lines
1.3 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
###
|
|
# File: common-functions.sh
|
|
# Project: bin
|
|
# File Created: Tuesday, 6th October 2022 9:30:00 pm
|
|
# Author: Josh.5 (jsunnex@gmail.com)
|
|
# -----
|
|
# Last Modified: Tuesday, 6th October 2022 9:30:00 pm
|
|
# Modified By: Josh.5 (jsunnex@gmail.com)
|
|
###
|
|
|
|
wait_for_x() {
|
|
# Wait for X server to start
|
|
# (Credit: https://gist.github.com/tullmann/476cc71169295d5c3fe6)
|
|
MAX=60 # About 30 seconds
|
|
CT=0
|
|
while ! xdpyinfo >/dev/null 2>&1; do
|
|
sleep 0.50s
|
|
CT=$(( CT + 1 ))
|
|
if [ "$CT" -ge "$MAX" ]; then
|
|
LOG "FATAL: $0: Gave up waiting for X server $DISPLAY"
|
|
exit 11
|
|
fi
|
|
done
|
|
}
|
|
|
|
|
|
get_nvidia_gpu_id() {
|
|
# Fech NVIDIA GPU device (if one exists)
|
|
if [ "${NVIDIA_VISIBLE_DEVICES:-}" == "all" ]; then
|
|
gpu_select=$(nvidia-smi --format=csv --query-gpu=uuid 2> /dev/null | sed -n 2p)
|
|
elif [ -z "${NVIDIA_VISIBLE_DEVICES:-}" ]; then
|
|
gpu_select=$(nvidia-smi --format=csv --query-gpu=uuid 2> /dev/null | sed -n 2p)
|
|
else
|
|
gpu_select=$(nvidia-smi --format=csv --id=$(echo "${NVIDIA_VISIBLE_DEVICES:-}" | cut -d ',' -f1) --query-gpu=uuid | sed -n 2p)
|
|
if [ -z "$gpu_select" ]; then
|
|
gpu_select=$(nvidia-smi --format=csv --query-gpu=uuid 2> /dev/null | sed -n 2p)
|
|
fi
|
|
fi
|
|
echo ${gpu_select}
|
|
}
|