Check for nvidia-smi after the environment variables in the MIG hook

Author: 3XX0

conf/hooks/extra/50-mig-config.sh

@@ -19,7 +19,7 @@ shopt -s lastpipe
 
 source "${ENROOT_LIBRARY_PATH}/common.sh"
 
-common::checkcmd grep nvidia-smi
+common::checkcmd grep
 
 tac "${ENROOT_ENVIRON}" | grep "^NVIDIA_" | while IFS='=' read -r key value; do
     [ -v "${key}" ] || export "${key}=${value}"
@@ -32,6 +32,8 @@ if [[ "${NVIDIA_VISIBLE_DEVICES}" =~ "MIG-" ]] || [[ "${NVIDIA_VISIBLE_DEVICES}"
     exit 0
 fi
 
+common::checkcmd nvidia-smi
+
 nvsmi_args=("--query-gpu=mig.mode.current" "--format=csv,noheader")
 
 if [ "${NVIDIA_VISIBLE_DEVICES}" != "all" ]; then