Zengtudor/windows_in_docker
1
0
Fork 0
mirror of https://github.com/dockur/windows.git synced 2025-06-23 16:07:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #1 from Steel-skull/Steel-skull-patch-2

Browse Source 
Update gpu-switch.sh
This commit is contained in:
Steel 2024-11-13 22:10:44 -06:00 committed by GitHub
commit e8fa68ed39
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
gpu-switch.sh
View File

@ -35,21 +35,24 @@ convert_to_pci_address() {
local device="$1" local device="$1"
local gpu_address="" local gpu_address=""
if [[ "$device" =~ ^[0-9]+$ ]]; then if [[ "$device" =~ ^[0-9]+$ || "$device" =~ ^GPU-.*$ ]]; then
# Convert GPU index to PCI address # Convert GPU index or UUID to PCI address
gpu_address=$(nvidia-smi --id=$device --query-gpu=gpu_bus_id --format=csv,noheader 2>/dev/null | tr -d '[:space:]') gpu_address=$(nvidia-smi --id="$device" --query-gpu=gpu_bus_id --format=csv,noheader 2>/dev/null | tr -d '[:space:]')
elif [[ "$device" =~ ^GPU-.*$ ]]; then
# Handle UUID
gpu_address=$(nvidia-smi --id=$device --query-gpu=gpu_bus_id --format=csv,noheader 2>/dev/null | tr -d '[:space:]')
else else
# Direct PCI address provided # Direct PCI address provided
gpu_address=$device gpu_address="$device"
fi
# Check for valid output
if [ -z "$gpu_address" ]; then
error_exit "Failed to get PCI address for device: $device"
fi fi
# Standardize format # Standardize format
echo "$gpu_address" | sed 's/0000://' | sed 's/\./:/g' echo "$gpu_address" | sed -e 's/0000://' -e 's/\./:/g'
} }
get_gpu_addresses() { get_gpu_addresses() {
# Split devices by comma # Split devices by comma
IFS=',' read -ra DEVICES <<< "$NVIDIA_VISIBLE_DEVICES" IFS=',' read -ra DEVICES <<< "$NVIDIA_VISIBLE_DEVICES"