无法让PyTorch与TensorBoard一起工作

我正在按照这个教程设置PyTorch(通过conda安装的v1.3.0版本)与TensorBoard https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html#

但是在这一步时

from torch.utils.tensorboard import SummaryWriter# default `log_dir` is "runs" - we'll be more specific herewriter = SummaryWriter('runs/fashion_mnist_experiment_1')

我不断遇到错误

---------------------------------------------------------------------------ModuleNotFoundError                       Traceback (most recent call last)C:\ProgramData\Anaconda3\envs\fastai_v1\lib\site-packages\torch\utils\tensorboard\__init__.py in       1 try:----> 2     from tensorboard.summary.writer.record_writer import RecordWriter  # noqa F401      3 except ImportError:ModuleNotFoundError: No module named 'tensorboard.summary'; 'tensorboard' is not a packageDuring handling of the above exception, another exception occurred:ImportError                               Traceback (most recent call last)c:\Users\matt\Documents\code\playground\tensorboard.py in ----> 1 from torch.utils.tensorboard import SummaryWriter      2       3 # default `log_dir` is "runs" - we'll be more specific here      4 writer = SummaryWriter('runs/fashion_mnist_experiment_1')C:\ProgramData\Anaconda3\envs\fastai_v1\lib\site-packages\torch\utils\tensorboard\__init__.py in       2     from tensorboard.summary.writer.record_writer import RecordWriter  # noqa F401      3 except ImportError:----> 4     raise ImportError('TensorBoard logging requires TensorBoard with Python summary writer installed. '      5                       'This should be available in 1.14 or above.')      6 from .writer import FileWriter, SummaryWriter  # noqa F401ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed. This should be available in 1.14 or above.

有谁有建议吗?


回答:

错误日志中提到,

ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed. This should be available in 1.14 or above.

因此,当它试图导入TensorBoard时,由于在搜索路径中缺少TensorBoard,无法完成导入。你可以安装最新版本(不指定版本号),如:

$ conda install -c conda-forge tensorboard

此外,你可能还需要安装protobuf:

$ conda install -c conda-forge protobuf

这些安装应该能解决ImportError问题。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注