pytorch ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed

今天想要试验在pytorch上运行tensorboard时发现报错如下:

Traceback (most recent call last):
  File "/home/../try.py", line 80, in <module>
    from torch.utils.tensorboard import SummaryWriter
  File "/home/../anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/utils/tensorboard/__init__.py", line 4, in <module>
    raise ImportError('TensorBoard logging requires TensorBoard with Python summary writer installed. '
ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed. This should be available in 1.14 or above.

解决办法

情况 1

如果还报有没有No module named ‘tensorboard’ 的错,则说明是由于没有安装tensorboard引起的,类似一下这种报错

Traceback (most recent call last):
  File "/home/../anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/utils/tensorboard/__init__.py", line 2, in <module>
    from tensorboard.summary.writer.record_writer import RecordWriter  # noqa F401
ModuleNotFoundError: No module named 'tensorboard'

所以解决办法是安装tensorboard即可

我是在anaconda环境下的,因此只需如下命令:

conda install tensorboard

如果不是在anaconda环境下,可以pip install tensorboard安装

情况 2

如果确认安装了tensorboard 依然报错,应该和版本有关
可以尝试如下命令解决

pip install future