首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >AVX512未在英特尔Tensorflow上显示

AVX512未在英特尔Tensorflow上显示
EN

Stack Overflow用户
提问于 2022-07-19 00:40:03
回答 1查看 393关注 0票数 1

我有一个Windows 11计算机与第11代英特尔核心i7-1185G7,它支持SSE4.1,SSE4.2,AVX,AVX2和AVX512。这台电脑没有GPU。

我用Python3.10创建了一个conda环境,并运行了pip install intel-tensorflow。根据文档,命令pip install intel-tensorflow-avx512只应该在Linux平台上使用。它提到在PIP车轮上自动使用和启用AVX512:

所有英特尔TensorFlow二进制文件都是使用oneAPI深层神经网络库(oneDNN)进行优化的,该库将在性能关键操作中自动使用AVX2或AVX512F FMA等CPU指令,这是基于您的机器上支持的用于Windows和Linux的指令集。

但是,当我启动一个使用TensorFlow的新项目时,将显示以下消息:

代码语言:javascript
运行
复制
I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX AVX2
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.

因此,我不确定TensorFlow是否使用AVX512作为默认指令集。

问题

  1. 如何检查TensorFlow是否确实在使用AVX512?
  2. 如果TensorFlow没有使用AVX512,我如何才能强迫它使用?这是一个应该向英特尔报告的错误吗?
  3. 当在CPU上使用AVX512训练TensorFlow模型时,与AVX和AVX2相比,TensorFlow真的值得吗?
EN

回答 1

Stack Overflow用户

发布于 2022-07-24 01:28:39

这可能并不理想,但您可以使用intel-tensorflow-a 512包作为测试,尝试WSL并运行TF。

它应该是默认的TF窗口包以及(不需要使用avx512 pip包),但我现在确认。会尽快回复你的。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/73029954

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档