首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >tensorflow-gpu比tensorflow慢

tensorflow-gpu比tensorflow慢
EN

Stack Overflow用户
提问于 2020-04-01 14:37:26
回答 1查看 102关注 0票数 0

我用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。

代码语言:javascript
复制
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = ""
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2020-04-01 14:45:27

从TensorFlow 2.1开始,GPU支持也在tensorflow包中可用,而不仅仅是tensorflow-gpu;如果你使用import tensorflow as tf,它会默认使用GPU,如果它找到了一个;我个人会首先卸载tensorflow-gpu,只留下普通的tensorflow包。

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

https://stackoverflow.com/questions/60964933

复制
相关文章

相似问题

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