我想通过CLI在远程访问的GPU服务器上训练一个OpenNLP模型。我熟悉在训练pytorch模型时使用GPU,但意识到我不确定这在OpenNLP中如何操作,因为它是用Java编写的。如果我在GPU上训练,OpenNLP会使用GPU吗?
具体来说,我在想我们训练pytorch模型时常用的代码片段:
if torch.cuda.is_available(): dev = "cuda:0"
谁能解释一下这在Java的OpenNLP库中是如何工作的?有没有类似于这行代码的等效内容?
我还在远程GPU服务器上使用这个Docker镜像来运行CLI:https://hub.docker.com/r/casetext/opennlp/dockerfile
我认为我还需要修改Dockerfile才能使用GPU,但我想知道是否需要先对OpenNLP代码做些什么,以便无论是否使用Docker容器都能实现这一点。
回答:
Apache OpenNLP不支持在GPU上进行训练。训练只能在CPU上进行。