Question

我目前正在Amazon EC2实例上运行，并尝试在Jupyter Notebook（Python 2.7）中运行以下代码：

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data/")
images, labels = mnist.train.images, mnist.train.labels
images = images.reshape((55000,28,28))
np.savez_compressed("MNIST_data/train", images=images, labels=labels)

但是，我收到以下错误：

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-4-de596f1c663d> in <module>()
      1 
----> 2 from tensorflow.examples.tutorials.mnist import input_data
      3 mnist = input_data.read_data_sets("MNIST_data/")
      4 images, labels = mnist.train.images, mnist.train.labels
      5 images = images.reshape((55000,28,28))

ImportError: No module named tensorflow.examples.tutorials.mnist

到目前为止，我还没有安装任何库以使TensorFlow代码正常工作（而且我不确定为什么我不需要安装任何东西）。如何安装此示例集？

Answer 1

我建议使用安装了TensorFlow（以及其他MachineLearning包）的AWS DeepLearning AMI。

深度学习Amazon AMI的AMI ID如下：
  us-east-1：ami-4b44745d
  us-east-2：ami-305d7c55
  us-west-2：ami-296e7850
  欧 - 西1：ami-d36386aa
  ap-southeast-2：ami-52332031
  ap-northeast-1：ami-b44050d3
  ap-northeast-2：ami-1523fc7b

发布用于DW框架的标签/分支：
  MXNet：v0.10.0标签
   TensorFlow：v1.1.0标记
  Theano：rel-0.8.2标签
  Caffe：rc5标签
  Caffe2：v0.7.0
  Keras：1.2.2标签
  CNTK：v2.0.rc1标签
  火炬：主分支

请参阅description了解更多相关信息

Answer 2

嗯，以下，

from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("MNIST_data", one_hot=True)

应该工作，我只是仔细检查。也许您安装了一个非常旧的TensorFlow版本？您可以通过

查看版本

import tensorflow as tf
tf.__version__

我不记得input_data何时添加到tensorflow.examples.tutorials.mnist但它应该适用于Tf 1.1和1.2

在EC2实例中安装TensorFlow示例

2 个答案: