Add deep learning models into Image of IIVR Project.
94
作者：魏文应 时间：2024-01-12
视频修复算法在下面环境中运行：
| 条目 | 型号/版本 | 备注 |
|---|---|---|
| 硬件 | 华为昇腾Atlas800-9000服务器 | NPU910B、鲲鹏ARM64架构CPU |
| 系统 | Ubuntu20.04 | 内核必须为：Linux guest 5.4.0-26-generic |
| 物理机NPU驱动 | Ascend-hdk-910-npu-driver_23.0.rc2_linux-aarch64.run Ascend-hdk-910-npu-firmware_6.4.12.1.241.run Ascend-cann-toolkit_6.3.RC2_linux-aarch64.run Ascend-docker-runtime_5.0.RC2_linux-aarch64.run | 版本号必须严格对应 |
| Python | Python3.8.18 | 要求3.8版本 |
| 容器内CANN | Ascend-cann-toolkit_6.3.RC2_linux-aarch64.run | 相当于NPU的CUDA |
| 神经网络框架 | PyTorch2.0.1 | Ascend torch_npu适配版 |
使用下面命令进行启动：
image=nuvic/iivr-model:0.0.1-base-ascendpytorch2.0.1-ubuntu20.04-arm64
container=iivr_model
# 使用镜像创建容器
docker run \
-itd -u 0 \
--privileged \
--ipc=host \
--net=host \
--device=/dev/davinci6 \
--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 /home/installFiles:/home/installFiles \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
--name=${container} ${image} /bin/bash
然后物理机上，执行下面命令，登录容器：
docker exec -it ${container} /bin/bash
执行下面命令，查看模型是否已经被复制到容器内：
ls /root/work/iivr/user_opt/
Content type
Image
Digest
sha256:7d29488d6…
Size
4.1 GB
Last updated
over 2 years ago
docker pull nuvic/iivr-model:0.0.1-base-ascendpytorch2.0.1-ubuntu20.04-arm64