tensorflow-gpu-nightly包

时间:2018-02-16 22:13:49

标签: python tensorflow deep-learning gpu

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

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

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

1 个答案:

答案 0 :(得分:2)

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

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

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

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