簡體   English   中英

帶Bazel的Tensorflow:忽略了AVX嗎?

[英]Tensorflow with bazel: avx ignored?

我正在嘗試手動編譯張量流以利用avx的優勢。 不幸的是,我似乎特別忽略了這些說明。 我可能會補充說我正在Windows 10上工作。任何幫助將不勝感激。 謝謝。

我將以下命令用於bazel:

bazel build -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma --copt=-mfpmath=both -k //tensorflow/tools/pip_package:build_pip_package

用bazel編譯

-mavx等僅適用於GCC和Clang編譯器。 由於使用的是MSVC,因此應使用--copt=/arch:AVX --copt=/arch:AVX2 ,依此類推。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM