出错情况
在使用stressGAN时出现
Traceback (most recent call last):
File "train.py", line 27, in <module>
from tensorboardX import SummaryWriter
File "/home/wang/.local/lib/python3.5/site-packages/tensorboardX/__init__.py", line 5, in <module>
from .torchvis import TorchVis
File "/home/wang/.local/lib/python3.5/site-packages/tensorboardX/torchvis.py", line 10, in <module>
from .writer import SummaryWriter
File "/home/wang/.local/lib/python3.5/site-packages/tensorboardX/writer.py", line 16, in <module>
from .comet_utils import CometLogger
File "/home/wang/.local/lib/python3.5/site-packages/tensorboardX/comet_utils.py", line 7, in <module>
from .summary import _clean_tag
File "/home/wang/.local/lib/python3.5/site-packages/tensorboardX/summary.py", line 12, in <module>
from .proto.summary_pb2 import Summary
File "/home/wang/.local/lib/python3.5/site-packages/tensorboardX/proto/summary_pb2.py", line 5, in <module>
from google.protobuf.internal import builder as _builder
ImportError: cannot import name 'builder'
错误原因
因为当前下载的protobuf依赖缺少buidler.py这个文件,可以通过pip install --upgrade protobuf
去升级protobuf
解决方法
- 将缺失的builder.py文件下载到本地中
wget https://raw.githubusercontent.com/protocolbuffers/protobuf/main/python/google/protobuf/internal/builder.py .
- 将builder.py文件复制到缺失的目录下即可
cp ./builder /home/wang/.local/lib/python3.5/site-packages/google/protobuf/internal/