3 回答

TA貢獻(xiàn)1946條經(jīng)驗(yàn) 獲得超4個(gè)贊
pip install tensorflow
TensorFlow在啟動(dòng)時(shí)檢查它是否已使用CPU上可用的優(yōu)化進(jìn)行編譯。如果不包括優(yōu)化,TensorFlow將發(fā)出警告,例如不包括AVX、AVX 2和FMA指令。
什么是SSE4.2和AVX?
弗林分類學(xué)中的一類并行計(jì)算機(jī)。它描述了具有多個(gè)處理元素的計(jì)算機(jī),這些處理元素同時(shí)對(duì)多個(gè)數(shù)據(jù)點(diǎn)執(zhí)行相同的操作。因此,這類機(jī)器利用數(shù)據(jù)級(jí)的并行性,而不是并發(fā)性:有同時(shí)(并行)計(jì)算,但在給定時(shí)刻只有一個(gè)進(jìn)程(指令)。
這些SSE4.2和AVX如何改進(jìn)TF任務(wù)的CPU計(jì)算
如何使用這兩個(gè)庫使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
添加回答
舉報(bào)