Update patch.sh

Find Path to nvidia-smi. 
For example on gentoo is: "/opt/bin/nvidia-smi"
pull/34/head
3PO 5 years ago committed by GitHub
parent 8191f86d41
commit 10c12bf821
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -71,8 +71,9 @@ declare -A object_list=(
["415.25"]='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

Loading…
Cancel
Save