Using my GPU for BERT nlp analysis

Hello. I am trying to get my GPU to exectute with my NLP analysis, but as of right now, only ram and CPU are being Utilized, making the process painstakingly slow. Can somebody provide a step by step guide on how to implement this?

@ajbond or @Artem, would you mind having a look?

Thanks!
Steffen

Hello @rickybordon

In the latest Knime version (4.7.*) Python environment is bundled along with all necessary dependencies. But you can also try to do is to create an environment, just go to the settings of Redfield BERT nodes and create a new GPU environment by clicking on New environment button, then select create GPU environemnt.

Otherwise you can you try to install all necessary Python packages yourself: you can find the list of packages in this blog post: BERT Text Classification for Everyone | KNIME

2 Likes

This topic was automatically closed 90 days after the last reply. New replies are no longer allowed.