Update patch.sh

Find Path to nvidia-smi. For example on gentoo is: "/opt/bin/nvidia-smi"
5 years ago · 1000312b62
parent 6f533954f6
commit 1000312b62
1 changed files with 2 additions and 1 deletions
--- a/patch.sh
+++ b/patch.sh
@ -73,8 +73,9 @@ declare -A object_list=(
    ["415.27"]='libnvcuvid.so'
 )

+NVIDIA_SMI="$(which nvidia-smi)"

-if ! driver_version=$(/usr/bin/nvidia-smi --query-gpu=driver_version --format=csv,noheader,nounits | head -n 1) ; then
+if ! driver_version=$($NVIDIA_SMI --query-gpu=driver_version --format=csv,noheader,nounits | head -n 1) ; then
    echo 'Something went wrong. Check nvidia driver'
    exit 1;
 fi