diff --git a/nvidia_entrypoint.sh b/nvidia_entrypoint.sh index f3a6a54ff3..f10c2673d5 100755 --- a/nvidia_entrypoint.sh +++ b/nvidia_entrypoint.sh @@ -69,7 +69,8 @@ else fi fi -if ! cat /proc/cpuinfo | grep flags | sort -u | grep avx >& /dev/null; then +# avx instructions are only available on x86, so don't check for aarch64 +if ! cat /proc/cpuinfo | grep flags | sort -u | grep avx >& /dev/null && [[ "aarch64" != $(uname -m) ]] ; then echo echo "ERROR: This container was built for CPUs supporting at least the AVX instruction set, but" echo " the CPU detected was $(cat /proc/cpuinfo |grep "model name" | sed 's/^.*: //' | sort -u), which does not report"