seems like not using CUDA
Closed this issue · 2 comments
sWAGgygoat commented
It seems like my script running for getting alignments failed to use CUDA so it's running very slow.
How should I assure using CUDA devices? plus I have joined CUDA_VISIBLE_DEVICES.
pdufter commented
Hi @sWAGgygoat, did you set device="cuda" when instantiating the SentenceAligner class?
sWAGgygoat commented
Thank you for your kindly reminder!
I found it later and problem solved!
发自我的iPhone
…------------------ Original ------------------
From: Philipp Dufter ***@***.***>
Date: Mon,Jul 26,2021 4:12 PM
To: cisnlp/simalign ***@***.***>
Cc: 程一迅 ***@***.***>, Mention ***@***.***>
Subject: Re: [cisnlp/simalign] seems like not using CUDA (#26)
Hi @sWAGgygoat, did you set device="cuda" when instantiating the SentenceAligner class?
—
You are receiving this because you were mentioned.
Reply to this email directly, view it on GitHub, or unsubscribe.