- 博客(2)
- 收藏
- 关注
RSS订阅原创 atc工具转换模型权重文件
/ model即你的.onnx文件，framework即onnx选项，output为输出的.om文件路径，soc_version为你的卡版本，比如我这里就是Ascend310P3。// 由于本人使用的是pytorch训练得到的.pth或者.pth.tar文件，需要先进行.pth向.onnx文件的转换。// 之后就可以传输到服务器上进行atc转换，用resnet50的模型进行测试。// 想要使用华为的推理卡进行推理，就需要有.om的权重文件。// 以下为python的转换代码。// 输出.onnx文件。
2024-02-28 15:57:06
539
原创 在Ascend 昇腾310P3推理卡上的镜像环境搭建
/ 使用 cat /etc/os-release查看服务器的操作系统，比如centos或者ubuntu，本人使用的服务器是华为的openEuler。// 接下来的目标是搭建docker，然后在容器上进行推理，同时需要将npu挂载到容器上。// 使用docker ps来查看容器情况，容器id前4位可以表示创建的容器。// 使用vim ~/.bashrc打开文件，加载驱动so，添加以下内容。// 之后就进入了容器，然后查看容器中可以使用的davinci设备。
2024-02-05 12:17:01
2165
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人