训练OpenNLP模型时如何使用GPU？

Question

我想在远程访问的GPU服务器上使用CLI训练一个OpenNLP模型。我很熟悉在训练pytorch模型时利用GPU，但我意识到我不确定这在openNLP中如何工作，因为它是用java编写的。如果我在GPU上进行训练，openNLP会利用GPU吗？

具体来说，我想到的是我们在训练pytorch模型时使用的这个熟悉的代码片段。

if torch.cuda.is_available():  
  dev = "cuda:0" 

谁能告诉我这段代码在java OpenNLP库中是如何工作的? 有没有与这行代码对应的地方？

我也在使用这个docker镜像在我的远程GPU服务器上运行CLI。https:/hub.docker.comrcasetextopennlpdockerfile。

我相信我还需要修改dockerfile才能使用GPU，但我想知道，无论我的docker容器使用情况如何，我是否需要先对openNLP代码做其他事情来完成这个任务。

Answer 1

Apache OpenNLP不支持在GPU上进行训练。只能在CPU上进行训练。