首页 文章

tensorflow-gpu-nightly包的Tensorflow CPU警告

提问于
浏览
0

当我启动tensorflow会话时,我收到以下错误:您的CPU支持此TensorFlow二进制文件未编译使用的指令:AVX AVX2

我已经为Windows安装了GPU nightly版本并安装了CUDA GPU工具包9.0 . 这是一个CPU警告,不应该因为我有GPU并使用它来运行tensorflow模型 .

以下是我在训练模型时的GPU使用情况(任务管理器):GPU Usage link - task manager

1 回答

  • 1

    Tensorflow二进制文件总是有CPU代码,无论它是否支持GPU . 此警告将显示在任何具有预先构建的Tensorflow二进制文件的合理新CPU上 .

    支持GPU的二进制文件包含用于Tensorflow OP的GPU内核,因此可以将许多计算量大的OP卸载到GPU . 但是总有一些OP没有GPU内核,而且最重要的是,总有一些代码只能在CPU上运行才能启动程序 .

    预先构建的Tensorflow二进制文件不是使用较新CPU支持的指令构建的,以便能够(几乎)在任何地方运行 .

    使用二进制文件来利用CPU必须提供的所有功能的唯一方法是从源代码构建,无论是本机编译还是使用适当的目标进行交叉编译 . 只有这样,这些警告才会消失 .

相关问题