tensorflow - 在 AWS SageMaker 上安装 Keras 和 Tensorflow

Question

我正在尝试将 Keras 下载到我在 AWS SageMaker 上的笔记本实例。下面列出了代码和错误或警告:

from keras.models import Sequential #Sequential Models
from keras.layers import Dense #Dense Fully Connected Layer Type
from keras.optimizers import SGD #Stochastic Gradient Descent Optimizer
from keras.callbacks import EarlyStopping
from keras.wrappers.scikit_learn import KerasClassifier

错误:

ModuleNotFoundError: No module named 'tensorflow'

然后我尝试下载 Tensorflow:

!pip install tensorflow

安装完成，注意以下几点:

Installing collected packages: wrapt, tensorflow
Found existing installation: wrapt 1.10.11
    Cannot uninstall 'wrapt'. It is a distutils installed project and thus we cannot accurately determine which files belong to it which would lead to only a partial uninstall.
    You are using pip version 10.0.1, however version 19.1.1 is available.
    You should consider upgrading via the 'pip install --upgrade pip' command.

然后我尝试卸载 wrapt 但仍然有同样的问题。有没有人有同样的问题？这是 SageMaker 相关的问题吗？我也尝试运行以下行，但没有改变:

from sagemaker.tensorflow import TensorFlow

Answer 1

尝试这个:

!pip install tensorflow -t ./

它将在您的当前目录中安装 tensorflow