Tensorflow DCGAN的踏坑之路(二)

    换了一台笔记本电脑,暗影精灵4,GTX1070的显卡,可以安装tensorflow-gpu版本了,依然是1.13.1版本,网上查了下需要装cuda和cudnn,但具体装哪个版本,也是查了半天,后来在这个博客中找到了,感谢博主_沥川往事,在这个blog(https://blog.csdn.net/yuejisuo1948/article/details/81043962)中列出来tensorflow CUDA cudnn 版本对应关系,这里借用一下图。

    刚好有1.13.1版本的,python也是3.7版本,所以下载的cuda是10.0版本,cuDNN是7.4版本,具体如下:

cuda_10.0.130_win10_network.execudnn-10.0-windows10-x64-v7.4.1.5.zip

    安装好了之后,再跑MNIST,速度真是天壤之别,当然,可能我的thinkpad里面2G显存的显卡没用,用的只是CPU版本,所以运算速度差别太大,使用GPU版本,跑1个epoch,只用了2分钟左右时间。然后接下来就是要跑另外一个数据集,celebA了,今天太晚了,等后续再更新吧。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值