有无GPU运行深度学习mnist数据集时间对比

#硬件配置

*本人用的是 lenovo小新锐7000笔记本,cpu是intel -core -i5-7300Q
四核,内存4G,机械硬盘320G,双显卡,集成intel® HD Graphics630
独显 GeForce GTX 1050,这块显卡位宽128bit, 显存2G,有640个cuda核心。
##系统环境:
win7下,安装Anaconda3, 之后安装Python3.6 , Cuda9.0,cudnn9.0 tensorflow1.14.0, Pycharm专业版 ,训练该数据集用了两层的卷积神经网络,训练2万步,在不开启GPU加速的情况下用时59分06秒;切换到GPU虚拟环境下用时5分12秒。看来没有GPU,实践深度学习确实有难度呀。
附代码如下:
*

//import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data

def weight_variable(shape):
    initial = tf.truncated_normal(shape, stddev=0.1)
    return tf.Variable(initial)

def bias_variable(shape):
    initial = tf.constant(0.1, shape=shape)
    return tf.Variable(initial)

def conv2d(x, W):
    return tf.nn.conv2d(x, W, strides=[1, 1, 1, 1], padding='SAME')

def max_pool_2x2(x):
    return tf.nn.max_pool(x, ksize=[1, 2, 2
  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值