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

這是一次傻逼的經歷,突然報如下錯誤,我以爲是環境壞掉了,重裝環境搞了半天,猜猜最後是什麼原因?

Traceback (most recent call last):
  File "/home/leerw/.local/lib/python3.6/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.summary'; 'tensorboard' is not a package

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "../util/torch_nn.py", line 19, in <module>
    from pytorch_lightning import LightningModule
  File "/home/leerw/.local/lib/python3.6/site-packages/pytorch_lightning/__init__.py", line 31, in <module>
    from .trainer import Trainer
  File "/home/leerw/.local/lib/python3.6/site-packages/pytorch_lightning/trainer/__init__.py", line 28, in <module>
    from .trainer import Trainer
  File "/home/leerw/.local/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 15, in <module>
    from pytorch_lightning.loggers import LightningLoggerBase
  File "/home/leerw/.local/lib/python3.6/site-packages/pytorch_lightning/loggers/__init__.py", line 78, in <module>
    from .tensorboard import TensorBoardLogger
  File "/home/leerw/.local/lib/python3.6/site-packages/pytorch_lightning/loggers/tensorboard.py", line 8, in <module>
    from torch.utils.tensorboard import SummaryWriter
  File "/home/leerw/.local/lib/python3.6/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.

最後查到原因是我在該目錄下創建了一個tensorboard.py文件,但是什麼都沒寫。看看pytorch背地裏幹了什麼勾當?

try:
    from tensorboard.summary.writer.record_writer import RecordWriter  # noqa F401
except ImportError:
    raise ImportError('TensorBoard logging requires TensorBoard with Python summary writer installed. '
                      'This should be available in 1.14 or above.')
from .writer import FileWriter, SummaryWriter  # noqa F401

怪不得我在別的文件夾執行

from torch.utils.tensorboard import SummaryWriter

沒有問題

該問題過於傻逼,寫下來紀念一下。

如果你也遇到了這個問題,但是不像我一樣傻逼的話,下面這個鏈接可能會幫到你:
https://github.com/pytorch/pytorch/issues/20140

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章