diff --git a/nvidia-smi b/nvidia-smi index 2cce86c..5ecc5be 100755 --- a/nvidia-smi +++ b/nvidia-smi @@ -1,12 +1,12 @@ -#!/usr/bin/bash - -for a in $* -do - case $a in - vgpu) - export LD_PRELOAD="/opt/vgpu_unlock-rs/target/release/libvgpu_unlock_rs.so" - ;; - esac -done - +#!/usr/bin/bash + +for a in $* +do + case $a in + vgpu) + export LD_PRELOAD="/opt/vgpu_unlock-rs/target/release/libvgpu_unlock_rs.so" + ;; + esac +done + exec /usr/bin/nvidia-smi $@ \ No newline at end of file