johanels/photoprism
1
0
Fork 0
mirror of https://github.com/photoprism/photoprism.git synced 2025-12-12 08:44:04 +01:00
Code Issues Packages Projects Releases Wiki Activity

AI: Update build target and install-tensorflow.sh for AVX512 #222 #4922

Browse Source 
Signed-off-by: Michael Mayer <michael@photoprism.app>
This commit is contained in:
Michael Mayer
2025-04-06 10:15:34 +02:00
parent a3456637cd
commit 332e57d59f
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/tensorflow/Makefile
View File

@@ -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)

4
scripts/dist/install-tensorflow.sh vendored
View File

@@ -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