search

PaddlePaddle / PaddleHelix

Bio-Computing Platform Featuring Large-Scale Representation Learning and Multi-Task Deep Learning “螺旋桨”生物计算工具集
Other
1.02k stars 225 forks source link

关于使用XPU #191

Closed EBGU closed 2 years ago

EBGU commented 2 years ago

您好！ 我目前在使用百度昆仑的R200 XPU服务器， 我想了解一下HelixFold是否支持在XPU上训练和推理，谢谢！

xiaoyao4573 commented 2 years ago

感谢对我们的关注，昆仑团队给我们反馈正在适配算子，有进一步消息会及时给你反馈。

xiaoyao4573 commented 2 years ago

我们的团队尝试在XPU进行训练和推理，验证可以跑通，直接使用对应的包即可。

EBGU commented 2 years ago

请问怎么set devices呀? 具体的说，我希望实现a）指定4张XPU种的某一张运行类似torch.device('cuda:0')这样 b）将一个长序列放到4张卡上跑（毕竟一张卡显存只有16GB） 以及XPU能跑OpenMM relax吗？

zhxfl commented 2 years ago

使用这个产出包替换原来的Paddle产出包即可使用单卡推理 wget https://baidu-kunlun-public.su.bcebos.com/KunlunPaddle/paddle-2.3-models/paddlepaddle_xpu-2.3.0-cp37-cp37m-linux_x86_64.whl