寒武纪mlu-270在docker内安装驱动

一、事先准备

1.装机

  • 将mlu270插到主板的显卡位
    寒武纪mlu-270在docker内安装驱动_第1张图片
  • 安装lspci
apt-get install pciutils
  • 开机查看是否安装成功
lspci | grep cabc

结果

查到270板卡

2.安装包

需要的安装包,从购买商处获取,存储在~/neuwareBasic文件夹
新版本安装包中,只需要获取cn-toolkit.deb即可

3.创建docker

#/bin/bash
#--device=/dev/video0
export MY_CONTAINER="Cambricon-Test-270-v1.4.0-lhd-v1212"
num=`sudo docker ps -a|grep "$MY_CONTAINER"|wc -l`
echo $num,$MY_CONTAINER
if [ 0 -eq $num ];then
  sudo xhost +
  sudo docker run \
    -e DISPLAY=unix$DISPLAY \
    --device /dev/cambricon_dev0 \
    --net=host \
    --pid=host \
    -v /sys/kernel/debug:/sys/kernel/debug \
    -v /tmp/.X11-unix:/tmp/.X11-unix \
    -v $PWD/neuwareBasic:/workdir \
    -w /workdir \
    -it \
    --privileged \
    --name $MY_CONTAINER \
    cambricon/test/ubuntu:v6.0 \
    /bin/bash
else
  sudo docker start $MY_CONTAINER
  #sudo docker attach $MY_CONTAINER
  sudo docker exec -w /workdir -ti $MY_CONTAINER /bin/bash
fi

/workdir/文件夹就是neuwareBasic文件夹

二、安装驱动

  • 首先使用dpkg安装驱动包
dpkg -i neuware-mlu270-driver-dkms_4.9.2_all.deb
  • 之后安装工具包,命令为
dpkg -i cntoolkit_1.7.3-2.ubuntu16.04_amd64.deb
  • 之后会解压为若干deb包,包括cncc,cnml等,找到安装位置/var/cntoolkit-1.7.3,继续输入dpkg -i *.deb
dpkg: error processing package neuware-mlu270-driver-dkms (--configure):

过程中会反复报错,即包之间存在依赖关系,可以再次输入命令安装
此时输入cnmon,如果显示出mlu信息,则证明成功
寒武纪mlu-270在docker内安装驱动_第2张图片
如果遇到其他软件包未安装,apt-get install 安装即可。

  • 添加环境变量
    cncc等在/usr/local/neuware/bin文件夹下,修改~/.bashrc文件,在末尾添加
export PATH=/usr/local/neuware/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/neuware/lib64 # 链接文件库
export NEUWARE_HOME=/usr/local/neuware

生效

source ~/.bashrc

输入cncc查看是否成功
成功查到命令

  • mlu文件编程方式参照这里

你可能感兴趣的:(tvm,硬件架构)