1. 首页
  2. 编程语言
  3. Python
  4. Tensorflow_MNIST_GPU

Tensorflow_MNIST_GPU

上传者: 2019-05-03 14:16:15上传 PY文件 3.09KB 热度 41次
电脑的显存只有2G,原始的GPU版的MNIST代码中,测试样本没有进行batch处理,而是将10000个测试样本直接存入显存(因为此处用的GPU运行程序),所以导致显存不够用而报错。此处提供一个batch处理的MNIST代码,可帮助读者进行测试。
用户评论
码姐姐匿名网友 2019-05-03 14:16:15

因为没有VPN就只能从CSDN上下载了,很好。