milvus向量库看起来对GPU没有依赖,可以拉取最新的torch镜像或其他DCU镜像,docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.4.1-ubuntu22.04-dtk25.04.1-py3.10,然后跟随官方quickstart部署milvus试试。
需要注意的是pip install pymilvus[model]时会安装依赖onnxruntime,需在安装pymilvus之前手动安装dcu版的onnxruntime,下载地址:https://download.sourcefind.cn:65024/4/main/onnxruntime/DAS1.6

暂无数据