飞桨对昆仑2代芯片的支持

飞桨自2.3rc版本起支持在昆仑2代芯片上(R200,R300)运行,经验证的模型训练的支持情况如下:

训练支持

可进行单机单卡/单机多卡训练的模型,如下所示:

模型 领域 编程范式 可用的CPU类型 单机单卡支持 单机多卡支持
ResNet50 图像分类 动态图 X86(Intel) 支持 -
MobileNet_v3 图像分类 动态图 X86(Intel) 支持 -
UNet 图像分割 动态图 X86(Intel) 支持 -
Yolov3-DarkNet53 目标检测 动态图 X86(Intel) 支持 -
SSD-ResNet34 目标检测 动态图 X86(Intel) 支持 支持
OCR-DB 文字检测 动态图 X86(Intel) 支持 -
Bert-Base NLP 静态图 X86(Intel) 支持 支持
Transformer NLP 静态图 X86(Intel) 支持 支持
GPT-2 NLP 动态图 X86(Intel) 支持 -
DeepFM 推荐 动态图 X86(Intel) 支持 -
Wide&Deep 推荐 动态图 X86(Intel) 支持 -

模型放置在飞桨模型套件中,作为github.com/PaddlePaddle下的独立repo存在,git clone下载即可获取所需的模型文件:

领域 套件名称 分支/版本
图像分类 PaddleClas develop
目标检测 PaddleDetection develop
图像分割 PaddleSeg develop
NLP PaddleNLP develop
OCR PaddleOCR dygraph
推荐 PaddleREC master
  • 注:支持基于Kermel Primitive算子的昆仑2代芯片支持,点击这里