2017-07-07 129 views
0

我目前在Amazon EC2實例上運行,我試圖運行在一個Jupyter筆記本電腦下面的代碼(Python 2.7版)的例子:安裝TensorFlow在EC2實例

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代碼的工作(我不完全確定爲什麼我不需要安裝任何東西)。我如何安裝這個示例集?

回答

1

嗯,下面,

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

+0

嘿塞巴斯蒂安 - 感謝您的快速答覆。運行版本檢查將返回「沒有名爲Tensorflow的模塊」錯誤。事實證明,我的Amazon實例預裝了Theano,但不包含TensorFlow。你知道我會如何在我的實例中使用Python 2.7安裝TensorFlow嗎? – Parseltongue

+1

我通常從Ubuntu實例中使用自己的AMI,但我不知道它們是如何設置的。但是,如果你幸運的話,就可以像從命令行執行'pip install tensorflow-gpu'一樣簡單。或者,您可以嘗試其他答案中提到的DL AMI。或者,如果您想自己在AWS上設置DL實例/ AMI,請參閱以下步驟:https://sebastianraschka.com/pdf/books/dlb/appendix_h_cloud-computing.pdf – Sebastian

1

我會建議使用AWS深度學習AMI附帶TensorFlow(其他MachineLearning軟件包)已安裝。

The AMI Ids for the Deep Learning Amazon Linux AMI are the following:
us-east-1: ami-4b44745d
us-east-2: ami-305d7c55
us-west-2: ami-296e7850
eu-west-1: ami-d36386aa
ap-southeast-2: ami-52332031
ap-northeast-1: ami-b44050d3
ap-northeast-2: ami-1523fc7b

Release tags/Branches used for the DW Frameworks:
MXNet : v0.10.0 tag
TensorFlow : v1.1.0 tag
Theano : rel-0.8.2 tag
Caffe : rc5 tag
Caffe2: v0.7.0
Keras : 1.2.2 tag
CNTK : v2.0.rc1 tag
Torch : master branch

description更多地瞭解它