Как я могу использовать тензорную доску с pytorch? - PullRequest
Новая
       7

Как я могу использовать тензорную доску с pytorch?

1 голос
/ 02 февраля 2020

Я следовал учебнику по тензорной доске pytorch: https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html.

Но я даже не могу запустить из-за следующей ошибки: 

from torch.utils.tensorboard import SummaryWriter

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
~/apps/anaconda3/envs/torch/lib/python3.7/site-packages/torch/utils/tensorboard/__init__.py in <module>
      1 try:
----> 2     from tensorboard.summary.writer.record_writer import RecordWriter  # noqa F401
      3 except ImportError:

ModuleNotFoundError: No module named 'tensorboard'

During handling of the above exception, another exception occurred:

ImportError                               Traceback (most recent call last)
<ipython-input-4-c8ffdef1cfab> in <module>
----> 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')

~/apps/anaconda3/envs/torch/lib/python3.7/site-packages/torch/utils/tensorboard/__init__.py in <module>
      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.

Я установил Pytorch 1.14 через Конда. Я должен установить что-то еще?

1 Ответ

5 голосов
/ 02 февраля 2020

Этот урок, вероятно, должен дать вам знать, что вам нужно установить тензорную доску. Взгляните на документацию по тензорной доске pytorch , которая объясняет, что вам нужно сначала установить тензорную доску.

В основном вы можете установить тензорную доску, используя 

pip install tensorboard

, а затем запустить тензорную доску сервер, запустив 

tensorboard --logdir=runs

Каталог runs - это то место, куда будет писать ваш автор сводок, и сервер тензорной доски, с которого он будет читать, чтобы узнать, что нужно визуализировать.

...