如何使用TensorFlow GPU？

8 浏览
0 Comments

如何使用TensorFlow GPU？

如何在Python 3.6 x64中使用TensorFlow GPU版本而不是CPU版本？

import tensorflow as tf

Python正在使用我的CPU进行计算。

我能注意到这一点，因为我遇到了一个错误：

您的CPU支持的指令，这个TensorFlow二进制文件没有编译使用：AVX2

我已经安装了tensorflow和tensorflow-gpu。

如何切换到GPU版本？

0