tflearn无法读取mnist数据

时间:2017-02-18 10:54:38

标签: tensorflow tflearn

我正在尝试使用tensorflow进行卷积神经网络。我正在使用tflearn。我已从this site

下载了数据
Traceback (most recent call last):
File "C:/Users/Sukumar/Python/ML/cnn_tflearn.py", line 7, in <module>
X, Y, test_x, test_y = mnist.load_data(one_hot=True)
File "C:\Users\Sukumar\Miniconda3\lib\site-packages\tflearn\datasets\mnist.py", line 15, in load_data
mnist = read_data_sets("mnist/", one_hot=one_hot)
File "C:\Users\Sukumar\Miniconda3\lib\site-packages\tflearn\datasets\mnist.py", line 161, in read_data_sets
train_images = extract_images(local_file)
File "C:\Users\Sukumar\Miniconda3\lib\site-packages\tflearn\datasets\mnist.py", line 50, in extract_images
buf = bytestream.read(rows * cols * num_images)
File "C:\Users\Sukumar\Miniconda3\lib\gzip.py", line 274, in read
return self._buffer.read(size)
TypeError: only integer scalar arrays can be converted to a scalar index

1 个答案:

答案 0 :(得分:0)

MNIST文件将与支持GPU的TensorFlow版本一起使用,该版本使用NVIDIA cuDNN,这里有一篇关于安装How to verify CuDNN installation?的文章。