300T算力切分问题
收藏回复举报
300T算力切分问题
t('forum.solved') 已解决
发表于2023-09-28 16:05:49
0 查看

物理机环境

  • OS:Centos7.6

  • driver:23.0.rc2

  • fi'rmware:6.4.12.1.241

  • toolkit:6.3.RC2

创建vnpu

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)

运行测试脚本

true

我要发帖子