search

undertheseanlp / underthesea

Underthesea - Vietnamese NLP Toolkit
http://undertheseanlp.com
GNU General Public License v3.0
1.38k stars 273 forks source link

Does underthesea support GPU accelerator ? #531

Closed hduc-le closed 2 years ago

hduc-le commented 2 years ago

Hi guys, I'm working on a project using the underthesea library for dependency parsing tasks. Everything is okay when I work with CPU mode, however, if I activate GPU mode, I will receive an error message as below:

RuntimeError: 'lengths' argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor.

So, I wonder does the underthesea support GPU mode?

rain1024 commented 2 years ago

Please give me more information about your environments and your commands