PyTorch使用tensorboard的SummaryWriter报错

PyTorch使用tensorboard可以显示网络运行情况,但偶尔使用SummaryWriter时遇到Segmentation fault错误。
利用python3的faulthandler,可定位到出错的代码行,具体操作有两种方式如下:
(1) 在代码中写入faulthandler

import faulthandler
# 在import之后直接添加以下启用代码即可
faulthandler.enable()
# 后边正常写你的代码

(2)直接通过命令行来启用,运行时添加-X faulthandler参数即可:

python -X faulthandler your_script.py

错误代码:

import xxx
from torch.utils.tensorboard import SummaryWriter
import yyy

log_dir = './run_logs'
self.writer = SummaryWriter(log_dir)

报错如下:

 File "/opt/conda/lib/python3.7/site-packages/tensorboard/compat/__init__.py", line 45 in tf
  File "/opt/conda/lib/python3.7/site-packages/tensorboard/lazy.py", line 50 in load_once
  File "/opt/conda/lib/python3.7/site-packages/tensorboard/lazy.py", line 97 in wrapper
  File "/opt/conda/lib/python3.7/site-packages/tensorboard/lazy.py", line 65 in __getattr__
  File "/opt/conda/lib/python3.7/site-packages/tensorboard/summary/writer/event_file_writer.py", line 72 in __init__
  File "/opt/conda/lib/python3.7/site-packages/torch/utils/tensorboard/writer.py",line 66 in __init__
  File "/opt/conda/lib/python3.7/site-packages/torch/utils/tensorboard/writer.py",line 256 in _get_file_writer
  File "/opt/conda/lib/python3.7/site-packages/torch/utils/tensorboard/writer.py", line 225 in __init__


经过查询,发现极有可能是导包(from torch.utils.tensorboard import SummaryWriter)顺序导致的问题。参考链接:[TensorBoard] The different order of import SummaryWriter may cause Segmentation fault

将from torch.utils.tensorboard import SummaryWriter顺序调前,可以顺利导包和实例化了。

import xxx
from torch.utils.tensorboard import SummaryWriter
import yyy

改为

from torch.utils.tensorboard import SummaryWriter
import xxx
import yyy

相关推荐

  1. PyTorch使用tensorboardSummaryWriter

    2024-06-07 22:48:02       32 阅读
  2. Pytorch:torch.utils.tensorboard.SummaryWriter

    2024-06-07 22:48:02       37 阅读
  3. pytorchTensorBoard使用

    2024-06-07 22:48:02       29 阅读
  4. pytorch GPU cuda 使用 整理

    2024-06-07 22:48:02       29 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-07 22:48:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-07 22:48:02       101 阅读
  3. 在Django里面运行非项目文件

    2024-06-07 22:48:02       82 阅读
  4. Python语言-面向对象

    2024-06-07 22:48:02       91 阅读

热门阅读

  1. DeepSort整体流程梳理及匈牙利算法解析

    2024-06-07 22:48:02       29 阅读
  2. PyCharm中快速搭建Python虚拟环境的指南

    2024-06-07 22:48:02       33 阅读
  3. 《effective c++》学习笔记

    2024-06-07 22:48:02       26 阅读
  4. npm yarn 更换国内源以及node历史版本下载地址

    2024-06-07 22:48:02       38 阅读
  5. leetcode 279.完全平方数

    2024-06-07 22:48:02       32 阅读
  6. 使用OpenCV进行简单图像分割的3个步骤

    2024-06-07 22:48:02       28 阅读
  7. ES 面试手册

    2024-06-07 22:48:02       29 阅读
  8. 2024河南高考作文ChatGPT

    2024-06-07 22:48:02       36 阅读