【算能】在Docker中调用PCIe卡

2023-10-13 22:52

文章标签 docker 调用 pcie 算能

本文主要是介绍【算能】在Docker中调用PCIe卡，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

开发需求，需要在centos下开发对应的内容

首先拉取docker 镜像

docker pull centos:centos7

然后在空白的centos容器下使用PCIe卡，这个部分特别提醒，需要挂载/dev的这个目录，才能读到内容，故而创建docker的命令

docker run --restart always --privileged -v /dev:/dev -td -v <工作目录>:/workspace --name <指定容器名字> <镜像名> bash
对于我们的情况，我们使用下面的命令
docker run --restart always --privileged -v /dev:/dev -td -v /home/xuanjie.xiao/dockerfiles:/workspace --name sdk1run centos:centos7 bash
docker run --restart always --privileged -v /dev:/dev -td -v /home/xuanjie.xiao/SDKTEST:/workspace --name s1run centos:centos7 bash

相关创建docker的命令解析，请查看对应的链接内容

docker exec -it <上述容器名> bash

因为使用的是centos

故而需要安装对应的支撑包：

sudo yum install -y epel-release
sudo yum install -y dkms
sudo yum install -y ncurses*

然后安装libsophon的包

安装libsophon：
sudo  rpm -ivh sophon-driver-0.4.6-1.x86_64.rpm
sudo  rpm -ivh sophon-libsophon-0.4.6-1.x86_64.rpm
sudo  rpm -ivh --force sophon-libsophon-dev-0.4.6-1.x86_64.rpm
在终端执行如下命令，或者登出再登入当前用户后即可使用bm-smi等命令：
source /etc/profile

安装sophon-mw

sudo rpm -ivh sophon-mw-sophon-ffmpeg_0.6.0_amd64.rpm sophon-mw-sophon-ffmpeg-dev_0.6.0_amd64.rpm
sudo rpm -ivh sophon-mw-sophon-opencv-abi0_0.6.0_amd64.rpm sophon-mw-sophon-opencv-abi0-dev_0.6.0_amd64.rpm

安装sophon-sail之前应该安装cmake3

yum -y install cmake3

安装libsophon的包

dpkg -i ./sophon-libsophon_<version>_amd64.deb ./sophon-libsophon-dev_<version>_amd64.deb

检查环境是否正常

# 导入运行环境
source /etc/profile.d/libsophon-bin-path.sh
# 检查容器内运行环境是否正常
bm-smi

打开终端执行 lspci | grep Sophon 检查卡是否能够被识别。正常情况下应该输出如下信息：101:00.0 Processing accelerators: Bitmain Technologies Inc. BM1684, Sophon Series Deep Learning Accelerator (rev 01)
若PCIe加速卡没有被系统正常识别，则需要首先排除故障，通常引起PCIe加速卡未被正常识别的可能原因有：PCIe加速卡在插糟中没有插紧；检查插卡的槽位是否是标准的X16槽位，X8槽位的功率支持通常最大只有45W，不建议使用；PCIe加速卡从PCIe直接供电，不需要外接电源，若连接了外接电源，可能导致卡不能被正常识别；三芯片以上PCIe加速卡需要足够的散热条件，若风道和风量不能符合PCIe加速卡的被动散热要求，则需要通过BIOS将风扇转速设置到足够大或者加装额外的风扇进行散热。

这篇关于【算能】在Docker中调用PCIe卡的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！