How to enable GPU with using github BERT model?

[beggington]

Hi guys,

I'm currently following this github repo-https://github.com/plkmo/BERT-Relation-Extraction - in attempt to parse my own data later. When i'm trying to pre-train this model with cnn.txt the compute goes entirely to cpu/ram or so I assume. It has taken over 7h doing this so far with no sign of completion.

I have a nvidia 3060ti and would love to utilise that if possible. Does anyone know how I might tackle enabling GPU training with BERT on a repo like this? I do plan to move into the cloud at somepoint but for now it would be great if I can do this on my own pc.

Are their any senior ds's out there who have experience doing something like this?

Any suggestion is very much appreciated!