1 从NV的GPU迁移到DCU
1.1. 构建DCU基础环境
参考第一部分 构建DCU基础环境, 完成 DCU 基础环境构建
1.2. 替换深度学习算法包
对于依赖cuda的深度学习算法包,需要替换为光合开发者社区的版本;
开发者社区:https://developer.hpccube.com/tool

手动下载其中对应算法包的whl文件到您的服务器,安装轮子:
参考如下步骤:
注意: 替换算法包的时候,算法包的版本需要和DTK对应;
pip 配置国内安装源
pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
pip install pip -U
安装轮子
pip install ***.whl
1.3. DCU 适配案例:
DCU开发者社区光源ModelZoo介绍(可以快速查询所需的DCU算法模型,根据Readme进行构建DCU环境,一键运行所需的算法模型):
https://sourcefind.cn/#/model-zoo/list
光源可以查询到基于多种DCU适配的各个AI技术领域的算法模型以及算法框架,如bert、yolo、resnet、Qwen、Llama、vllm、lmdpeloy等等;
Examples: