华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
OS:Centos7.6
driver:23.0.rc2
fi'rmware:6.4.12.1.241
toolkit:6.3.RC2
npu-smi set -t vnpu-mode -d 0 npu-smi set -t create-vnpu -i 5 -c 0 -f vir08 npu-smi set -t create-vnpu -i 5 -c 0 -f vir04 npu-smi set -t create-vnpu -i 5 -c 0 -f vir16 npu-smi info -t info-vnpu -i 5 -c 0 npu-smi set -t vnpu-cfg-recover -d 1
docker pull ascendhub.huawei.com/public-ascendhub/pytorch-modelzoo:23.0.RC2-1.11.0
docker run -it --ipc=host \ -u root \ --name=vnpu \ --device=/dev/vdavinci182:/dev/davinci182 \ --device=/dev/davinci_manager \ --device=/dev/devmm_svm \ --device=/dev/hisi_hdc \ -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \ -v /usr/local/Ascend/add-ons/:/usr/local/Ascend/add-ons/ \ -v /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi \ -v /usr/local/sbin/:/usr/local/sbin/ \ -v /var/log/npu/conf/slog/slog.conf:/var/log/npu/conf/slog/slog.conf \ -v /var/log/npu/slog/:/var/log/npu/slog \ -v /var/log/npu/profiling/:/var/log/npu/profiling \ -v /var/log/npu/dump/:/var/log/npu/dump \ -v /var/log/npu/:/usr/slog \ ascendhub.huawei.com/public-ascendhub/pytorch-modelzoo:23.0.RC2-1.11.0 \ /bin/bash
vim main.py
import torch import torch_npu a=torch.zeros(10) print(a) a=a.npu() print(a)
我要发帖子
物理机环境
OS:Centos7.6
driver:23.0.rc2
fi'rmware:6.4.12.1.241
toolkit:6.3.RC2
创建vnpu
拉取测试镜像
docker pull ascendhub.huawei.com/public-ascendhub/pytorch-modelzoo:23.0.RC2-1.11.0创建容器
创建测试脚本
vim main.py运行测试脚本