0%

寒武纪MLU220 开发环境Docker搭建

首先进入寒武纪开发主页并登录寒武纪账号,进入cambricon_pytorch docker 页面。由于本人只使用MLU220 进行边缘端推理,所以不需要在主机上安装MLU 驱动,因此可跳过第一步直接按照提示安装docker。本人在尝试按照页面说明时遇到了一些问题,在此记录一下。

若是初次使用docker,则建议将Ubuntu个人账户添加进docker组里,这样就避免每次使用时都要输入sudo了,该部分参考文章

1
2
3
4
5
## 将当前用户添加进docker组,并更新
sudo gpasswd -a user docker
newgrp docker

## 然后重启电脑才可永久生效

然后按照寒武纪提示,依次进行如下操作:

1
2
3
4
5
6
## 登录harbor
docker login cair.cambricon.com

## 输入用户名、API密钥(网页用户名下拉框中有API密钥选项)
Username: (username)
Password: (API密钥)

注意,这里网页提示使用docker pull命令下载相应的镜像文件,但经过本人尝试之后发现,在后续的docker run命令中会重复下载,因此跳过使用docker pull,直接使用docker run命令创建容器

1
docker run -it --name mlu220 -v /home/***/work:/work cair.cambricon.com/cambricon/cambricon_pytorch:ubuntu18.04_sdk_v1.7.0_pytorch_v0.15.0-2 /bin/bash

上述命令会自动下载镜像文件,并改名为“mlu220”,且将主机的“/home/*/work”映射至docker端的“/work”。

此时,可以查看新建的容器:

docker

启动mlu220容器并查看:

1
2
❯ docker start mlu220
mlu220

docker

最后进入docker并激活开发环境:

1
2
docker exec -it mlu220 /bin/bash
source torch/venv3/pytorch/bin/activate