Docker: Update scripts/dist/tensorflow-driver.sh

This commit is contained in:
Michael Mayer 2022-03-25 10:14:48 +01:00
parent c6753ca2bc
commit eaecedf4bb

View file

@ -1,7 +1,8 @@
#!/bin/bash
PATH="/usr/local/sbin:/usr/sbin:/sbin:/usr/local/bin:/usr/bin:/bin:/scripts"
CPU_DETECTED=$(/usr/bin/lshw -c processor -json 2>/dev/null)
CPU_DETECTED=$(lshw -c processor -json 2>/dev/null)
if [[ $(echo "${CPU_DETECTED}" | jq -r '.[].capabilities.avx2') == "true" ]]; then
echo "avx2"