我用tensorflow制作了神经网络,但tensorflow-gpu比cpu慢!
tensorflow 2.1和tensorflow-gpu 2.1的总运行(训练)时间分别为130秒和330秒
我的CPU是i7-第7代,而GPU是geforce-930M(笔记本电脑环境)这是因为我的GPU比CPU慢吗?如果是,我是否可以设置为仅在适当的情况下自动运行GPU?
(CUDA环境似乎设置适当,我还手动检查了tensorflow 2.1仅使用CPU,而tensorflow-gpu 2.1同时使用CPU和GPU。)
更新:我的神经网络的大小是64 x 32 x 16 x 1(可能不适合并行执行),在tensorflow 2.1中,我通过以下命令关闭了GPU。
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = ""发布于 2020-04-01 14:45:27
从TensorFlow 2.1开始,GPU支持也在tensorflow包中可用,而不仅仅是tensorflow-gpu;如果你使用import tensorflow as tf,它会默认使用GPU,如果它找到了一个;我个人会首先卸载tensorflow-gpu,只留下普通的tensorflow包。
https://stackoverflow.com/questions/60964933
复制相似问题