如何用SSE4.2和AVX指令編譯TensorFlow？這是運行腳本以檢查TensorFlow是否工作的消息：I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcublas.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcudnn.so.5 locally
I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcufft.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUDA library libcurand.so.8.0 locally
W tensorflow/core/platform/cpu_feature_guard.cc:95] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:95] The TensorFlow library wasn't compiled to use AVX instructions, but these are available on your machine and could speed up CPU computations.
I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:910] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero我注意到它提到了SSE4.2和AVX，1)什么是SSE4.2和AVX？2)這些SSE4.2和AVX如何改進TensorFlow任務的CPU計算。3)如何利用這兩個庫進行TensorFlow編譯？
3 回答
絕地無雙
TA貢獻1946條經驗 獲得超4個贊
pip install tensorflow
TensorFlow在啟動時檢查它是否已使用CPU上可用的優化進行編譯。如果不包括優化，TensorFlow將發出警告，例如不包括AVX、AVX 2和FMA指令。
什么是SSE4.2和AVX？
弗林分類學中的一類并行計算機。它描述了具有多個處理元素的計算機，這些處理元素同時對多個數據點執行相同的操作。因此，這類機器利用數據級的并行性，而不是并發性：有同時(并行)計算，但在給定時刻只有一個進程(指令)。
這些SSE4.2和AVX如何改進TF任務的CPU計算
如何使用這兩個庫使TensorFlow編譯？
bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mfpmath=both --copt=-msse4.2 --config=cuda -k //tensorflow/tools/pip_package:build_pip_package
添加回答
舉報
0/150
提交
取消