keras用cpu加速_Keras在GPU上的训练性能比CPU慢

我几天前才开始学习深度学习。

我试图用一个由猫和狗的照片组成的数据集训练CNN。

但问题是训练过程花费了太长的CPU时间,而且我可以看到我的CPU利用率在训练期间没有明显提高。而且训练过程非常缓慢。我以为是因为CPU的缘故,所以我继续安装CUDA和TensorFlow gpu,TensorFlow gpu似乎是作为显卡出现在控制台上的。Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 4620 MB memory) -> physical GPU (device: 0, name: GeForce RTX 2060, pci bus id: 0000:1c:00.0, compute capability: 7.5)

但是在GPU上的训练速度似乎比在CPU上慢。

而且在训练模型时它占用了所有的GPU内存,但是GPU的利用率在训练时似乎很低。你们能看看我做错什么了吗?我怎样才能加快速度呢?

系统规格:

CPU:Ryzen 2700x

GPU:ZOTAC RTX 2060

内存:DDR4 16GB 3000兆赫

我使用的代码:

^{pr2}$

速度:20/8000 [..............................] - ETA: 10:47 - loss: 0.0269 - acc: 0.9906

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值