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 GitHub
commit 10c12bf821
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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