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.exe和cudnn-10.0-windows10-x64-v7.4.1.5.zip。

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

你可能感兴趣的:(深度学习)