diff --git a/patch-fbc.sh b/patch-fbc.sh index 1ed2fa5..114acfc 100755 --- a/patch-fbc.sh +++ b/patch-fbc.sh @@ -111,8 +111,15 @@ patch_common () { exit 1 fi - if ! driver_version=$("$NVIDIA_SMI" --query-gpu=driver_version --format=csv,noheader,nounits | head -n 1) ; then - echo 'Something went wrong. Check nvidia driver' + cmd="$NVIDIA_SMI --query-gpu=driver_version --format=csv,noheader,nounits" + driver_versions_list=$($cmd) + ret_code=$? + driver_version=$(echo "$driver_versions_list" | head -n 1) + if [[ $ret_code -ne 0 ]] ; then + echo "Can not detect nvidia driver version." + echo "CMD: \"$cmd\"" + echo "Result: \"$driver_versions_list\"" + echo "nvidia-smi retcode: $ret_code" exit 1 fi diff --git a/patch.sh b/patch.sh index d4d5907..edf2a37 100755 --- a/patch.sh +++ b/patch.sh @@ -183,8 +183,15 @@ patch_common () { exit 1 fi - if ! driver_version=$("$NVIDIA_SMI" --query-gpu=driver_version --format=csv,noheader,nounits | head -n 1) ; then - echo 'Something went wrong. Check nvidia driver' + cmd="$NVIDIA_SMI --query-gpu=driver_version --format=csv,noheader,nounits" + driver_versions_list=$($cmd) + ret_code=$? + driver_version=$(echo "$driver_versions_list" | head -n 1) + if [[ $ret_code -ne 0 ]] ; then + echo "Can not detect nvidia driver version." + echo "CMD: \"$cmd\"" + echo "Result: \"$driver_versions_list\"" + echo "nvidia-smi retcode: $ret_code" exit 1 fi