调用summary方法使用导入tensorboard包,现在可以直接利用torch来导入,然而可以看到torch源文件中也是import tensorboard,所以需额外安装tensorboard。
import tensorboard
from distutils.version import LooseVersion
if not hasattr(tensorboard, "__version__") or LooseVersion(
tensorboard.__version__
) < LooseVersion("1.15"):
raise ImportError("TensorBoard logging requires TensorBoard version 1.15 or above")
del LooseVersion
del tensorboard
from .writer import FileWriter, SummaryWriter # noqa: F401
from tensorboard.summary.writer.record_writer import RecordWriter # noqa: F401
选择conda还是pip安装依据自己环境选择。