tensorflow初学者教程 - read_data_sets失败

时间:2015-12-01 01:02:10

标签: python numpy tensorflow mnist

我正在尝试在Linux上运行tutorial。我安装了gcccythonnumpysix

我可以导入数据但是解压缩似乎存在某种问题。

有人可以帮忙吗?

Python 2.7.3 (default, Jun 22 2015, 19:43:34)
[GCC 4.6.3] on linux2
Type "help", "copyright", "credits" or "license" for more information.   
>>> import g3doc.tutorials.mnist.input_data as input_data
>>> mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
Successfully downloaded train-images-idx3-ubyte.gz 9912422 bytes.
Extracting MNIST_data/train-images-idx3-ubyte.gz
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "g3doc/tutorials/mnist/input_data.py", line 175, in read_data_sets
    train_images = extract_images(local_file)
  File "g3doc/tutorials/mnist/input_data.py", line 60, in extract_images
    buf = bytestream.read(rows * cols * num_images)
  File "/usr/lib/python2.7/gzip.py", line 263, in read
    chunk = self.extrabuf[offset: offset + size]
TypeError: only integer scalar arrays can be converted to a scalar index
>>> mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)
Extracting MNIST_data/train-images-idx3-ubyte.gz
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "g3doc/tutorials/mnist/input_data.py", line 175, in read_data_sets
    train_images = extract_images(local_file)
  File "g3doc/tutorials/mnist/input_data.py", line 60, in extract_images
    buf = bytestream.read(rows * cols * num_images)
  File "/usr/lib/python2.7/gzip.py", line 263, in read
    chunk = self.extrabuf\[offset: offset + size]
TypeError: only integer scalar arrays can be converted to a scalar index

1 个答案:

答案 0 :(得分:8)

这似乎与Numpy的最新版本有关。 recent change将单元素数组视为用于索引的标量的错误。

我已对上游TensorFlow代码进行了相关更改,但在此期间您可以将this line in input_data.py(L45)编辑为以下内容(在行末添加[0]):< / p>

return numpy.frombuffer(bytestream.read(4), dtype=dt)[0]