返回列表
[AscendC][Pytorch2.1.0][910B] 910b-ascend_24.1.rc2-cann_8.0.rc2-py_3.10-ubuntu_22.04-aarch64_v1.3
PyTorch
Others
下载量:16
上传者:xMushroom
点赞
1
有改进之处
0
收藏
收藏
版本信息
Python版本:3.10
CANN版本:8.0.rc2
操作系统版本:ubuntu_22.04
使用说明
# 依赖【细看,版本对齐,不行就装】 + 驱动 Ascend-hdk-910b-npu-driver_24.1.rc2_linux-aarch64.run 链接:https://obs-whaicc-fae-public.obs.cn-central-221.ovaijisuan.com/cann/mindie/1.0.RC2/910b/Ascend-hdk-910b-npu-driver_24.1.rc2_linux-aarch64.run + 固件 Ascend-hdk-910b-npu-firmware_7.3.0.1.231.run 链接:https://obs-whaicc-fae-public.obs.cn-central-221.ovaijisuan.com/cann/mindie/1.0.RC2/910b/Ascend-hdk-910b-npu-firmware_7.3.0.1.231.run # 拉取镜像 + 修改 /etc/hosts 加入 58.48.42.198 swr.cn-central-221.ovaijisuan.com (一般不用) + 修改 /etc/docker/daemon.json 如下: ```json { "registry-mirrors": [ ...[没有就不写这个字段] ], "insecure-registries": [ "swr.cn-central-221.ovaijisuan.com" ] } ``` + 重启docker ```shell systemctl daemon-reload systemctl restart docker ``` + 拉取镜像 ```shell docker pull swr.cn-central-221.ovaijisuan.com/wh-aicc-fae/mindie:37131-ascend_24.1.rc2-cann_8.0.rc2-py_3.10-ubuntu_22.04-aarch64-mindie_1.0.RC2.01 ``` # 启动指令示例(自己加一下 【反斜杠】 我写了但不显示) ```shell docker run -it \ --name remote_01 \ --device=/dev/davinci0 \ --device=/dev/davinci1 \ --device=/dev/davinci_manager \ --device=/dev/devmm_svm \ --device=/dev/hisi_hdc \ -v /usr/local/dcmi:/usr/local/dcmi \ -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \ -v /usr/local/Ascend/driver/lib64/common:/usr/local/Ascend/driver/lib64/common \ -v /usr/local/Ascend/driver/lib64/driver:/usr/local/Ascend/driver/lib64/driver \ -v /etc/ascend_install.info:/etc/ascend_install.info \ -v /etc/vnpu.cfg:/etc/vnpu.cfg \ -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \ -v /home/aicc:/home/aicc \ -p 宿主机端口:22 \ swr.cn-central-221.ovaijisuan.com/wh-aicc-fae/cann:910b-ascend_24.1.rc2-cann_8.0.rc2-py_3.10-ubuntu_22.04-aarch64_v1.3 /bin/bash systemctl restart sshd # 不需要远程ssh容器内就不需要执行 ``` # 加载环境变量 ```shell # 自动加载,镜像启动或者用户登录会小卡10来秒,参阅 ~/.bashrc ``` # 测试代码 + 参阅 https://gitee.com/ascend/samples/tree/master/operator/AddCustomSample/KernelLaunch/AddKernelInvocationNeo + 注意: `samples/operator/HelloWorldSample/hello_world.cpp` 需要修改为: ```c++ #include "kernel_operator.h" using namespace AscendC; extern "C" __global__ __aicore__ void hello_world() { AscendC::printf("Hello World!!!\n"); } void hello_world_do(uint32_t blockDim, void *stream) { hello_world<<>>(); } ```
镜像地址
在其他AICC使用本镜像时,
1) 在本地arm宿主机通过docker pull 拉取上面镜像到本地(即执行docker pull remote_image_address)
2) 用docker tag 将局点信息和组织名替换成对应版本(即执行 docker tag local_image_address remote_image_address),
3) 用docker push 将修改后的镜像名称推送到局点的swr服务中(即执行docker push remote_image_address)
评论
若您有任何疑虑,您可以在这里反馈您的建议。
发布