Docker: Update scripts/dist/tensorflow-driver.sh
This commit is contained in:
parent
c6753ca2bc
commit
eaecedf4bb
1 changed files with 2 additions and 1 deletions
3
scripts/dist/tensorflow-driver.sh
vendored
3
scripts/dist/tensorflow-driver.sh
vendored
|
@ -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"
|
||||
|
|
Loading…
Reference in a new issue