我在Anaconda环境下使用Python (jupyter notebook)
操作系统: UbuntuTensorflow版本: 1.14.0Python版本: 3.6
https:/github.comtensorflowtensorboardissues1425。这也是同样的问题,但回答没有帮助。除了尝试重新安装Tensorflow之外,有没有其他的解决方法?或者我应该升级我的tensorflow版本?
from keras.callbacks import TensorBoard
tensorboard = TensorBoard(log_dir='graph_1', histogram_freq=0,
batch_size=512, write_graph=True, write_grads=False,
write_images=False, embeddings_freq=0, embeddings_layer_names=None,
embeddings_metadata=None, embeddings_data=None, update_freq='epoch')
抛出一个错误,比如。
ImportError Traceback (most recent call last)
~/anaconda3/lib/python3.6/site-packages/keras/callbacks.py in __init__(self, log_dir, histogram_freq, batch_size, write_graph, write_grads, write_images, embeddings_freq, embeddings_layer_names, embeddings_metadata, embeddings_data, update_freq)
744 import tensorflow as tf
--> 745 from tensorflow.contrib.tensorboard.plugins import projector
746 except ImportError:
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/__init__.py in <module>
40 from tensorflow.contrib import distribute
---> 41 from tensorflow.contrib import distributions
42 from tensorflow.contrib import estimator
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/distributions/__init__.py in <module>
43 from tensorflow.contrib.distributions.python.ops.distribution_util import tridiag
---> 44 from tensorflow.contrib.distributions.python.ops.estimator import *
45 from tensorflow.contrib.distributions.python.ops.geometric import *
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/distributions/python/ops/estimator.py in <module>
20
---> 21 from tensorflow.contrib.learn.python.learn.estimators.head import _compute_weighted_loss
22 from tensorflow.contrib.learn.python.learn.estimators.head import _RegressionHead
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/learn/__init__.py in <module>
92 # pylint: disable=wildcard-import
---> 93 from tensorflow.contrib.learn.python.learn import *
94 # pylint: enable=wildcard-import
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/learn/python/__init__.py in <module>
27 # pylint: disable=wildcard-import
---> 28 from tensorflow.contrib.learn.python.learn import *
29 # pylint: enable=wildcard-import
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/learn/python/learn/__init__.py in <module>
29 from tensorflow.contrib.learn.python.learn import datasets
---> 30 from tensorflow.contrib.learn.python.learn import estimators
31 from tensorflow.contrib.learn.python.learn import graph_actions
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/learn/python/learn/estimators/__init__.py in <module>
301 from tensorflow.contrib.learn.python.learn.estimators.constants import ProblemType
--> 302 from tensorflow.contrib.learn.python.learn.estimators.dnn import DNNClassifier
303 from tensorflow.contrib.learn.python.learn.estimators.dnn import DNNEstimator
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/learn/python/learn/estimators/dnn.py in <module>
33 from tensorflow.contrib.learn.python.learn import metric_spec
---> 34 from tensorflow.contrib.learn.python.learn.estimators import dnn_linear_combined
35 from tensorflow.contrib.learn.python.learn.estimators import estimator
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py in <module>
35 from tensorflow.contrib.learn.python.learn import metric_spec
---> 36 from tensorflow.contrib.learn.python.learn.estimators import estimator
37 from tensorflow.contrib.learn.python.learn.estimators import head as head_lib
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/learn/python/learn/estimators/estimator.py in <module>
51 from tensorflow.contrib.learn.python.learn.estimators._sklearn import NotFittedError
---> 52 from tensorflow.contrib.learn.python.learn.learn_io import data_feeder
53 from tensorflow.contrib.learn.python.learn.utils import export
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/learn/python/learn/learn_io/__init__.py in <module>
36 from tensorflow.contrib.learn.python.learn.learn_io.graph_io import read_keyed_batch_features_shared_queue
---> 37 from tensorflow.contrib.learn.python.learn.learn_io.numpy_io import numpy_input_fn
38 from tensorflow.contrib.learn.python.learn.learn_io.pandas_io import extract_pandas_data
~/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/learn/python/learn/learn_io/numpy_io.py in <module>
25
---> 26 from tensorflow.python.estimator.inputs.numpy_io import numpy_input_fn as core_numpy_input_fn
27 from tensorflow.python.util.deprecation import deprecated
~/anaconda3/lib/python3.6/site-packages/tensorflow/python/estimator/inputs/numpy_io.py in <module>
25
---> 26 from tensorflow_estimator.python.estimator.inputs import numpy_io
27
~/anaconda3/lib/python3.6/site-packages/tensorflow_estimator/python/estimator/inputs/numpy_io.py in <module>
25
---> 26 from tensorflow_estimator.python.estimator.inputs.queues import feeding_functions
27 from tensorflow.python.util.tf_export import estimator_export
~/anaconda3/lib/python3.6/site-packages/tensorflow_estimator/__init__.py in <module>
9
---> 10 from tensorflow_estimator._api.v2 import estimator
11 _names_with_underscore = []
~/anaconda3/lib/python3.6/site-packages/tensorflow_estimator/_api/v2/estimator/__init__.py in <module>
9
---> 10 from tensorflow_estimator._api.v2.estimator import experimental
11 from tensorflow_estimator._api.v2.estimator import export
~/anaconda3/lib/python3.6/site-packages/tensorflow_estimator/_api/v2/estimator/experimental/__init__.py in <module>
10 from tensorflow_estimator.python.estimator.canned.linear import LinearSDCA
---> 11 from tensorflow_estimator.python.estimator.canned.rnn import RNNClassifier
12 from tensorflow_estimator.python.estimator.canned.rnn import RNNEstimator
~/anaconda3/lib/python3.6/site-packages/tensorflow_estimator/python/estimator/canned/rnn.py in <module>
22
---> 23 from tensorflow.python.feature_column import dense_features
24 from tensorflow.python.feature_column import feature_column_lib as fc
ImportError: cannot import name 'dense_features'
During handling of the above exception, another exception occurred:
ImportError Traceback (most recent call last)
<ipython-input-83-0d8f7236c1d1> in <module>
16 reduce_lr_1 = ReduceLROnPlateau(monitor = 'val_loss', factor = 0.2, patience = 1, verbose = 1, min_delta = 0.0001)
17
---> 18 tensorboard_1 = TensorBoard(log_dir='graph_1', histogram_freq=0, batch_size=512, write_graph=True, write_grads=False, write_images=False, embeddings_freq=0, embeddings_layer_names=None, embeddings_metadata=None, embeddings_data=None, update_freq='epoch')
19
20 callbacks_1 = [checkpoint_1,earlystop_1,tensorboard_1,reduce_lr_1]
~/anaconda3/lib/python3.6/site-packages/keras/callbacks.py in __init__(self, log_dir, histogram_freq, batch_size, write_graph, write_grads, write_images, embeddings_freq, embeddings_layer_names, embeddings_metadata, embeddings_data, update_freq)
745 from tensorflow.contrib.tensorboard.plugins import projector
746 except ImportError:
--> 747 raise ImportError('You need the TensorFlow module installed to '
748 'use TensorBoard.')
749
ImportError: You need the TensorFlow module installed to use TensorBoard.
正如错误信息所显示的那样 ImportError: You need the TensorFlow module installed to use TensorBoard
.
看来你当前的Tensorflow(TF=1.14.0)安装存在问题。
要求你升级最新版本的Tensorflow (TF 2.2.0),如下图。
pip install tensorflow==2.2.0
#OR
pip install --upgrade tensorflow
我已经为TensorBoard启用了可视化,使用了 TF 2.2.0
在Jupyter Notebook中,如下所示
tf.keras.callbacks.TensorBoard
以下是完整的工作代码,以可视化模型精度和损失,如下所示
%load_ext tensorboard
import tensorflow as tf
import datetime, os
fashion_mnist = tf.keras.datasets.fashion_mnist
(x_train, y_train),(x_test, y_test) = fashion_mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0
def create_model():
return tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)),
tf.keras.layers.Dense(512, activation='relu'),
tf.keras.layers.Dropout(0.2),
tf.keras.layers.Dense(10, activation='softmax')
])
def train_model():
model = create_model()
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
logdir = os.path.join("logs", datetime.datetime.now().strftime("%Y%m%d-%H%M%S"))
tensorboard_callback = tf.keras.callbacks.TensorBoard(logdir, histogram_freq=1)
model.fit(x=x_train,
y=y_train,
epochs=5,
validation_data=(x_test, y_test),
callbacks=[tensorboard_callback])
train_model()
%tensorboard --logdir logs
输出。
Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/train-labels-idx1-ubyte.gz
32768/29515 [=================================] - 0s 0us/step
Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/train-images-idx3-ubyte.gz
26427392/26421880 [==============================] - 0s 0us/step
Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/t10k-labels-idx1-ubyte.gz
8192/5148 [===============================================] - 0s 0us/step
Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/t10k-images-idx3-ubyte.gz
4423680/4422102 [==============================] - 0s 0us/step
Epoch 1/5
1875/1875 [==============================] - 8s 4ms/step - loss: 0.4957 - accuracy: 0.8237 - val_loss: 0.4407 - val_accuracy: 0.8385
Epoch 2/5
1875/1875 [==============================] - 8s 4ms/step - loss: 0.3816 - accuracy: 0.8608 - val_loss: 0.3885 - val_accuracy: 0.8613
Epoch 3/5
1875/1875 [==============================] - 8s 4ms/step - loss: 0.3503 - accuracy: 0.8717 - val_loss: 0.3683 - val_accuracy: 0.8661
Epoch 4/5
1875/1875 [==============================] - 8s 4ms/step - loss: 0.3255 - accuracy: 0.8804 - val_loss: 0.3495 - val_accuracy: 0.8719
Epoch 5/5
1875/1875 [==============================] - 8s 4ms/step - loss: 0.3137 - accuracy: 0.8850 - val_loss: 0.3532 - val_accuracy: 0.8716
对于Jupyter用户 如果你已经将Jupyter和TensorBoard安装到同一个虚拟环境中 那你就可以开始了 如果你使用更复杂的设置,比如全局的Jupyter安装和不同Condavirtualenv环境的内核,那么你必须确保tensorboard二进制文件在Jupyter笔记本上下文的PATH上。一种方法是修改kernel_spec,将环境的bin目录预置到PATH上,如 此处.
更多详情请参考 此处