diff --git a/patch.sh b/patch.sh index ee9d638..faf86c9 100755 --- a/patch.sh +++ b/patch.sh @@ -285,16 +285,18 @@ get_supported_versions () { } patch_common () { + if [[ "$manual_driver_version" ]]; then + driver_version="$manual_driver_version" + + echo "Using manually entered Nvidia driver version: $driver_version" + else + NVIDIA_SMI="$(command -v nvidia-smi || true)" + if [[ ! "$NVIDIA_SMI" ]] ; then echo 'nvidia-smi utility not found. Probably driver is not installed.' exit 1 - fi - - if [[ "$manual_driver_version" ]]; then - driver_version="$manual_driver_version" - - echo "Using manually entered nvidia driver version: $driver_version" + else cmd="$NVIDIA_SMI --query-gpu=driver_version --format=csv,noheader,nounits" driver_versions_list=$($cmd) || ( @@ -308,7 +310,8 @@ patch_common () { driver_version=$(echo "$driver_versions_list" | head -n 1) echo "Detected nvidia driver version: $driver_version" - fi + + fi if ! check_version_supported "$driver_version" ; then echo "Patch for this ($driver_version) nvidia driver not found."