[PYTHON] Bei der Verwendung von Tensorboard mit Pytorch ist ein Fehler aufgetreten

Einführung

Ich habe versucht, Tensorboard mit Pytorch zu verwenden, habe es installiert, den Code geschrieben und ausgeführt, und es ist ein Fehler aufgetreten, sodass ich eine Problemumgehung dafür schreiben werde.

Installieren Sie Tensorboard

Da ich anaconda verwende, habe ich es mit dem folgenden Befehl installiert.

conda install tensorboard

Code

Importieren, um ein Diagramm in SummaryWriter zu zeichnen.

tensorboard.py

from torch.utils.tensorboard import SummaryWriter

Error

Ich habe den folgenden Importfehler erhalten.

ImportError: cannot import name 'SummaryWriter' from 'torch.utils.tensorboard' 

During handling of the above exception, another exception occurred:

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

Ansatz

Deinstallieren und neu installieren

Ich fragte mich, ob ich es nicht richtig installieren konnte, also löste ich es und installierte es erneut. Ich habe den gleichen Fehler bekommen. Ich habe mich hier verlaufen und es mit pip installiert. (Die Verwechslung zwischen Pip und Conda ist nicht sehr gut) Am Ende habe ich den gleichen Fehler bekommen.

Überprüfen Sie die Version von Pytorch

Da Tensorboard ab Version 1.2.0 auf Pytorch offiziell verwendet werden kann, habe ich vorerst die Version von Pytorch überprüft.

python

import torch
print(torch.__version__)

Es war *** 1.5.1 ***. Es scheint kein Pytorch-Problem zu sein.

Überprüfen Sie die Version von Tensorboard

Die Fehleranweisung besagt, dass die Tensorboard-Version 1.14 oder höher ist. Überprüfen Sie sie daher.

conda list

Es war *** 2.2.1 ***. Es scheint auch kein Tensorboard-Problem zu sein.

Datei umbenennen

Ich habe die Versionen von Pytorch und Tensorboard überprüft und es war okay, also muss ich es nicht mehr tun. Ich weiß nicht, was den Fehler verursacht.

Auf dem Dateinamen steht *** tensorboard.py ***. Liegt es daran, dass der Dateiname und das Modul kollidieren und nicht importiert werden können? Ich dachte.

Die Datei wurde in *** tb.py *** umbenannt. Und als ich es lief, bekam ich keine Fehler! !! !! !!

Zusammenfassung

Ich habe versucht, Tensorboard mit Pytorch zu verwenden und es installiert, aber ich habe einen Importfehler mit demselben Dateinamen wie das Modul erhalten. Halten Sie den Dateinamen und das Modul getrennt. Vermeiden Sie auch Verwechslungen zwischen Conda und Pip.