Open immortal-shuan opened 3 years ago
是否使用GPU取决于使用者有没有搭好GPU环境,跟模型提供者没任何关系。
在os里指定了GPU,但模型没有跑gpu。看网上说是加入这个,model = multi_gpu_model(roformer, 2)。
谢谢!
------------------ 原始邮件 ------------------ 发件人: "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.
是否使用GPU取决于使用者有没有搭好GPU环境,跟模型提供者没任何关系。