Skip to content

Commit 05d1b85

Browse files
committed
Try nvidia-smi commands for info
1 parent bda02cc commit 05d1b85

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

clip.sh

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -444,11 +444,8 @@ popd
444444
# Start processes
445445
if [ "$NVIDIA_HARDWARE_RENDERING" = "on" ]; then
446446
# Does not work on WSL2
447-
# GPU_BOARD=$(nvidia-xconfig --query-gpu-info | grep -i 'Name' | awk -F ': ' '{print $2}')
448-
# HACK: Hardcoded for now to workaround nvidia-xconfig not working on Replicate
449-
GPU_BOARD="Tesla T4"
450-
# GPU_PCI=$(nvidia-xconfig --query-gpu-info | grep BusID | awk '{print $4}')
451-
GPU_PCI="PCI:0:5:0"
447+
GPU_BOARD=$(nvidia-smi --query-gpu=name --format=csv,noheader)
448+
GPU_PCI=$(nvidia-smi --query-gpu=pci.bus_id --format=csv,noheader | awk -F ':' '{printf("PCI:%d:%d:0", "0x"$2, "0x"$3)}')
452449

453450
mkdir -p /etc/X11
454451
cat <<EOT > /etc/X11/xorg.conf

0 commit comments

Comments
 (0)