Tensorflow_MNIST_GPU

【文件属性】：

文件名称：Tensorflow_MNIST_GPU

文件大小：3KB

文件格式：PY

更新时间：2020-06-13 12:33:11

Tensorflow MNIST GPU

电脑的显存只有2G，原始的GPU版的MNIST代码中，测试样本没有进行batch处理，而是将10000个测试样本直接存入显存（因为此处用的GPU运行程序），所以导致显存不够用而报错。此处提供一个batch处理的MNIST代码，可帮助读者进行测试。