search

li3cmz / GRADE

GRADE: Automatic Graph-Enhanced Coherence Metric for Evaluating Open-Domain Dialogue Systems
57 stars 9 forks source link

cuda版本问题 #4

Closed 18166035475 closed 3 years ago

18166035475 commented 3 years ago

您好，是不是必须使用cuda 10.1呢，cpu上或者cuda11.1可以吗

James-Yip commented 3 years ago

您好，建议使用cuda10.1是因为我们实验时使用的是支持cuda10.1的pytorch版本，如果要使用cuda11.1的话，可以尝试更新到最新版本的pytorch。另外要在CPU上使用进行inference的话，需要删除掉代码中将model以及input移动到GPU的代码（建议还是使用GPU比较好）。