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