ZhuiyiTechnology / roformer-sim

SimBERT升级版(SimBERTv2)!
Apache License 2.0
439 stars 73 forks source link

怎么无法在GPU上运行 #4

Open immortal-shuan opened 3 years ago

bojone commented 3 years ago

是否使用GPU取决于使用者有没有搭好GPU环境,跟模型提供者没任何关系。

BEILOP commented 2 years ago

是否使用GPU取决于使用者有没有搭好GPU环境,跟模型提供者没任何关系。

在os里指定了GPU,但模型没有跑gpu。看网上说是加入这个,model = multi_gpu_model(roformer, 2)。

immortal-shuan commented 2 years ago

谢谢!

------------------ 原始邮件 ------------------ 发件人: "ZhuiyiTechnology/roformer-sim" @.>; 发送时间: 2021年11月29日(星期一) 中午12:09 @.>; @.**@.>; 主题: Re: [ZhuiyiTechnology/roformer-sim] 怎么无法在GPU上运行 (#4)

是否使用GPU取决于使用者有没有搭好GPU环境,跟模型提供者没任何关系。

在os里指定了GPU,但模型没有跑gpu。看网上说是加入这个,model = multi_gpu_model(roformer, 2)。

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android.