xuechaofei

具体应该怎么调整呢？

执行run_extractive_unified_ie.sh，GPU可以运行，但运行HugIE/api_test.py，一直都是使用CPU模式运行，找不到哪里设置GPU模式。运行前也执行了export CUDA_VISIBLE_DEVICES=0；请问是什么原因？

如果需要用GPU使用模型，需要在哪个位置修改代码？所有代码中，似乎没有传入GPU device参数的调用？

我们使用HugIE任务，训练了一些模型，但在使用模型时，找不到传入GPU device 的参数的代码位置，一直都只能用CPU在运行，效果很低。麻烦告知下如何传入GPU参数。