diff --git a/docker/tensorflow/Makefile b/docker/tensorflow/Makefile
index c48c17fc2..b97d66eec 100644
--- a/docker/tensorflow/Makefile
+++ b/docker/tensorflow/Makefile
@@ -19,5 +19,5 @@ libtensorflow-vnni:
 	bazel build -c opt //tensorflow:libtensorflow.so --copt=-O2 --copt=-mavx2 --copt=-mavxvnni --copt=-mfma --copt=-mfpmath=both --copt=-msse4.2
 	./create_archive.sh amd64-vnni $(TF_VERSION)
 libtensorflow-avx512:
-	bazel build -c opt //tensorflow:libtensorflow.so --copt=-O2 --copt=-mavx2 --copt=-mamx-avx512 --copt=-mfma --copt=-mfpmath=both --copt=-msse4.2
+	bazel build -c opt //tensorflow:libtensorflow.so --copt=-O2 --copt=-mavx2 --copt=-mavx512f --copt=-mavx512vnni --copt=-mfma --copt=-mfpmath=both --copt=-msse4.2
 	./create_archive.sh amd64-avx512 $(TF_VERSION)
diff --git a/scripts/dist/install-tensorflow.sh b/scripts/dist/install-tensorflow.sh
index 126724bd9..8a7a0e3c0 100755
--- a/scripts/dist/install-tensorflow.sh
+++ b/scripts/dist/install-tensorflow.sh
@@ -66,9 +66,9 @@ if [[ $TF_DRIVER == "auto" ]]; then
 
   CPU_DETECTED=$(lshw -c processor -json 2>/dev/null)
 
-  if [[ $(echo "${CPU_DETECTED}" | jq -r '.[0].capabilities.amx_tile') == "true" ]]; then
+  if [[ $(echo "${CPU_DETECTED}" | jq -r '.[0].capabilities.avx512_vnni') == "true" ]]; then
     TF_DRIVER="avx512"
-    echo "✅ Your CPU supports AMX-AVX512 Advanced Matrix Extensions"
+    echo "✅ Your CPU supports AVX512-VNNI Vector Neural Network Instructions"
     if [[ $(lshw -c display -json 2>/dev/null | jq -r '.[0].configuration.driver') == "nvidia" ]]; then
       echo "✅ You can optionally install the tensorflow-gpu version as an Nvidia GPU was detected"
     fi