BUG汇总

Pytorch BUGs

  1. AttributeError: module 'distutils' has no attribute 'version'
File "train.py", line 20, in <module>
    from torch.utils.tensorboard import SummaryWriter
  File "/group/dphi_algo_scratch_10/zhouping/anaconda3/envs/pt110/lib/python3.8/site-packages/torch/utils/tensorboard/__init__.py", line 4, in <module>
    LooseVersion = distutils.version.LooseVersion
AttributeError: module 'distutils' has no attribute 'version'

解决：
pip3 install setuptools==59.5.0

  1. RuntimeError: received 0 items of ancdata
    在dataloader加载数据时出现的错误，原因是pytorch多线程共享tensor是通过打开文件的方式实现的，而打开文件的数量是有限制的
    解决：修改多线程的tensor方式为file_system（默认方式为file_descriptor，受限于open files数量）
# 夹在train.py中即可
torch.multiprocessing.set_sharing_strategy('file_system')
posted @ 2022-03-11 10:34  小鸟飞飞11  阅读(485)  评论(0)  编辑  收藏  举报