ubuntu下安装transition_amr_parser
transition_amr_parser
是IBM公司开源的AMR paraing和AMR text-to-generation工具,在NLP领域中经常会用到,但是这个安装过程中可能会存在很多坑,transition_amr_parser的github主页安装教程不清晰,Python各种第三方包相互依赖,显卡驱动和torch版本匹配不上,导致出现大量的奇怪的错误。这里展示了我在ubuntu20.04、cuda为11.3环境下安装transition_amr_parser 0.3.3
/transition_amr_parser 0.5.2
/为例,两个版本都亲测安装成功。
下载对应版本的transition_amr_parser
可以通过transition_amr_parser的github主页进行下载你所需要的版本,如下图所示:
安装
新建一个虚拟的python环境,并切换到该环境下进行安装对应的依赖,首先安装torch
,由于安装过程中经常会提示torch
和cuda
版本不匹配,所以就从官网找到对应cuda
版本的torch
,我的版本为11.3,所以可以安装pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
,更多版本的可以查看Pytorch安装官网。一定要下载和自己版本一致的torch
,否则会出错,查看cuda
版本指令为nvcc -V
。
然后切换到下载的transition_amr_parser
目录中来,打开对应的README
文件,里面有详细的教程,以0.3.3为例,如下图所示:
可以省略torch
的安装,并且先切换到scripts/stack-transformer/requirements.txt文件中
,将torch
的安装注释掉,否则又会重新安装torch,将会出现各种问题。
等所有的安装完成后,就可以随意变化你的torch版本,但是一般高版本兼容低版本,所以也可以不切换到低版本torch
。
这也是针对AMR-IE事件抽取环境安装的整理。
测试自己是否安装成功,预训练模型下载地址,需要科学上网:
in_checkpoint = "path/to/your/checkpoint_best.pt"
from transition_amr_parser.stack_transformer_amr_parser import AMRParser
parser = AMRParser.from_checkpoint(in_checkpoint)
annotations = parser.parse_sentences([['The', 'boy', 'travels'], ['He', 'visits', 'places']])
print(annotations[0])
print(annotations[1])