无法使Pytorch与Tensorboard一起使用

Question

我正在学习本教程，以使用tensorboard https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html#设置pytorch（从conda到v1.3.0）

但要踩

from torch.utils.tensorboard import SummaryWriter

# default `log_dir` is "runs" - we'll be more specific here
writer = 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 package

During 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 F401

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

有人有什么建议吗？

Answer 1

除其他事项外，错误日志显示

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

因此，当它尝试导入TensorBoard时，无法这样做，因为它在搜索路径中丢失了它。您可以安装最新版本（不指定任何版本号），如下所示：

$ conda install -c conda-forge tensorboard

除此之外，您可能还需要安装protobuf：

$ conda install -c conda-forge protobuf

这些安装应修复ImportError。

Answer 2

我遇到了相同的错误，我通过以下步骤解决了该问题：

1. 从conda环境中删除了所有Tensorflow或Tensorboard的安装。
2. 然后通过激活相同的conda环境，键入“ pip install -U tb-nightly”
3. 然后输入“ pip install -U future”
4. 完成

Answer 3

我认为这是版本问题。

只需运行此：

pip install tensorboard==1.14.0

（不是pip install tensorboard == 1.14）

或仅安装包含tensorboard 1.14.0的tensoflow 1.14.0：

pip install tensorflow==1.14.0

这个版本的tensorflow在pytorch 1.2中对我有用。

Answer 4

我已经完成：

conda install -y tensorboard

之前没有问题，所以我不确定为什么这行不通。这是最简单的。

Answer 5

您尝试过Aim吗？它是为PyTorch构建的。记录和比较指标的最简单，最方便的工具（TensorBoard中不提供许多高级比较功能）。

Answer 6

我遇到了同样的错误，当时我的conda也没有工作，所以我选择使用tensorboardX，它和tensorboard几乎完全一样（还有它的操作）。< br/> 只需使用 pip install tensorboardX
安装它 您可以使用 from tensorboardX import SummaryWriter

导入它