xhw205 / GPLinker_torch

CMeIE/CBLUE/CHIP/实体关系抽取/SPO抽取
207 stars 14 forks source link

请问有成功运行代码的大佬提供一下环境版本信息吗? #12

Closed dr-GitHub-account closed 1 year ago

dr-GitHub-account commented 1 year ago

同服务器,之前别的代码环境torch==1.9.0+cu111,transformers==3.0.2,能跑起来。这里试了python3.8 - 3.9,PyTorch1.8.1 & 1.9.0,transformers4.9.2 & 3.0.2,跑不起来。要么提示distutils has no attribute 'version' (来源于main.py的line16,一般是需要降低setuptools版本),要么降了setuptools版本,从65降到59,又提示另一个错CUDA error: invalid device ordinal (来源于main.py的line39),应该是cuda版本又不对应了。刷包刷了半天,反反复复的...

dr-GitHub-account commented 1 year ago

自问自答一波,前一个问题和tensorboard有关,不装tensorboard或者降低setuptools版本可以解决这个问题。第二个问题可能不是CUDA版本问题,单纯就是main.py的device = torch.device("cuda:1")这里需要根据自己硬件情况改一下,之前看漏了。