Python如何使用tensorflow-cpu

Question

我成功地将tensorflow-cpu安装为pip install tensorflow-cpu

但是，当尝试导入包时，我在第一行收到此错误。

import tensorflow-cpu as tf
import numpy as np
from tensorflow.keras import Sequential
from tensorflow.keras.layers import Dense
l0 = Dense(units=1, input_shape=[1])
model = Sequential([l0])
model.compile(optimizer='sgd', loss='mean_squared_error')
xs = np.array([-1.0, 0.0, 1.0, 2.0, 3.0, 4.0], dtype=float)
ys = np.array([-3.0, -1.0, 1.0, 3.0, 5.0, 7.0], dtype=float)
model.fit(xs, ys, epochs=500)

ERROR Line 1: No module named tensorflow.

如何解决这个问题？

我没有更新的 GPU 或 NVIDIA 软件包，我的笔记本电脑不支持，因此建议使用 cpu 版本。

Answer 1

无论您安装哪种类型的 TensorFlow(CPU 或 GPU)，您都始终将 TensorFlow 作为 TensorFlow 导入，如下所示:

import tensorflow as tf