Update patch.sh

Find Path to nvidia-smi. 
For example on gentoo is: "/opt/bin/nvidia-smi"
This commit is contained in:
3PO 2019-01-19 14:18:54 +01:00 committed by Vladislav Yarmak
parent 6f533954f6
commit 1000312b62

View File

@ -73,8 +73,9 @@ declare -A object_list=(
["415.27"]='libnvcuvid.so' ["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' echo 'Something went wrong. Check nvidia driver'
exit 1; exit 1;
fi fi