首先进入寒武纪开发主页并登录寒武纪账号,进入cambricon_pytorch docker 页面。由于本人只使用MLU220 进行边缘端推理,所以不需要在主机上安装MLU 驱动,因此可跳过第一步直接按照提示安装docker。本人在尝试按照页面说明时遇到了一些问题,在此记录一下。
若是初次使用docker,则建议将Ubuntu个人账户添加进docker组里,这样就避免每次使用时都要输入sudo了,该部分参考文章:
1 | ## 将当前用户添加进docker组,并更新 |
然后按照寒武纪提示,依次进行如下操作:
1 |
|
注意,这里网页提示使用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”。
此时,可以查看新建的容器:
启动mlu220容器并查看:
1 | ❯ docker start mlu220 |
最后进入docker并激活开发环境:
1 | docker exec -it mlu220 /bin/bash |