安装部署

您可以通过 pip,docker,预编译二进制文件,从源代码构建等方式安装 BMF,选择您喜欢的一种即可。

BMF 支持 Python/C++/Go 跨语言调用,您可以通过任何语言调用用任何语言编写的模块。但是每种编程语言都有自己的要求。因此,如果您想通过 docker 安装之外的其它方法安装BMF并且有跨语言调用的需求,就需要设置相应的环境变量。详细说明请参阅以下各部分。

依赖

前置依赖

BMF 依赖一些库,您可以通过 apt、yum 或 brew 安装它们。一般来说,除了下面介绍的 docker 安装方法,其他方法均需要安装这些库。以下是我们测试过的系统以及安装方法,包括常规依赖、Python、FFmpeg、CUDA 工具包等。

  • 对于常规依赖,您需要在系统中安装所有 package。
  • 对于 Python,只有调用 Python API 或通过预编译安装包(在github的release产物)调用 Python 模块时才需要安装,目前的 Python 版本要求:Mac ARM 3.9 以上,Linux 和 Windows 3.7 以上。如果使用 pip 安装,则无需更改现有的的 Python 版本。
  • 对于 FFmpeg,我们目前支持 4.2 - 5.1 版本,将来可能会支持 6.x 版本。对于下面列出的 ubuntu、debian、CentOS:8 和 macOS,您可以使用 package 管理器安装。对于其他操作系统,您可能需要从源代码编译 FFmpeg。
ubuntu:20.04debian:11CentOS:8CentOS:7macos
regular dependency
apt update
apt install -y make \
    git pkg-config \
    libssl-dev \
    cmake binutils-dev \
    libgoogle-glog-dev \
    gcc g++ yasm nasm
apt update
apt install -y make \
    git pkg-config \
    libssl-dev cmake \
    libgoogle-glog-dev \
    binutils-dev gcc \
    g++ wget sudo bzip2 \
    libffi-dev zlib1g-dev
# install dependencies
dnf -y upgrade libmodulemd
dnf -y install glibc-langpack-en epel-release epel-next-release
dnf makecache
dnf update -y
dnf config-manager --set-enabled powertools
dnf -y install make git pkgconfig cmake3 openssl-devel binutils-devel gcc gcc-c++ glog-devel
yum -y install epel-release
yum -y install make git pkgconfig openssl-devel \
    cmake3 binutils-devel glog-devel gcc gcc-c++ \
    which wget
# rename cmake3 to cmake
ln -s $(which cmake3) $(dirname $(which cmake3))/cmake
brew install make git pkg-config openssl cmake glog
python
apt install -y python3.9 \
    python3-dev \
    python3-pip
apt install -y python3.9 \
    python3-dev \
    python3-pip
dnf -y install python39 python39-devel python39-pip

yum repo of CentOS:7 has no python3.9 libraries now, and needs to be built from source.

brew install python@3.9
FFmpeg
apt install -y \
    ffmpeg \
    libavcodec-dev \
    libavdevice-dev \
    libavfilter-dev \
    libavformat-dev \
    libavresample-dev \
    libavutil-dev \
    libpostproc-dev \
    libswresample-dev \
    libswscale-dev
apt install -y \
    ffmpeg \
    libavcodec-dev \
    libavdevice-dev \
    libavfilter-dev \
    libavformat-dev \
    libavresample-dev \
    libavutil-dev \
    libpostproc-dev \
    libswresample-dev \
    libswscale-dev
dnf install -y https://download1.rpmfusion.org/free/el/rpmfusion-free-release-8.noarch.rpm
dnf install -y https://download1.rpmfusion.org/nonfree/el/rpmfusion-nonfree-release-8.noarch.rpm
dnf install -y ffmpeg ffmpeg-devel

yum repo of CentOS:7 has no ffmpeg libraries now, please follow the below steps to build from source.

brew install ffmpeg@4
export DYLD_LIBRARY_PATH="/usr/local/opt/ffmpeg@4/lib:$DYLD_LIBRARY_PATH"

编译 Python

如上所述,如果您想通过预编译安装包调用 Python API 或 Python模块,又无法通过 package 管理器安装 Python3.9,则需要从源代码编译。

cd /opt
wget https://www.python.org/ftp/python/3.9.13/Python-3.9.13.tgz
tar xvf Python-3.9.13.tgz
cd Python-3.9.13
sudo ./configure --enable-optimizations --enable-shared
sudo make altinstall

编译 FFmpeg

另外,如上所述,从源代码编译 FFmpeg 是一个可选步骤,只有无法从 package 管理器安装时才需要。您可以自己编译,也可以使用我们提供的脚本(目前仅限 linux 和 macos):

git clone https://github.com/BabitMF/bmf bmf
cd bmf
./scripts/build_ffmpeg.sh x264 x265

GPU 依赖(仅限 Linux)

FFmpeg

如果您打算在 GPU 上使用 BMF,则需要安装支持 GPU 的 FFmpeg。根据上述 apt 或 dnf 安装的 FFmpeg 有 nvdec 或 nvenc,您可以使用 ffmpeg -encoders | grep nvenc 来检查。如果没有,则需要先编译 GPU 版本的 FFmpeg:

git clone https://github.com/BabitMF/bmf bmf
cd bmf
./scripts/build_ffmpeg.sh --device gpu

CUDA 工具包

如果您有 GPU 设备并且通过 ./scripts/build_ffmpeg.sh --device gpu 安装 FFmpeg, CUDA 工具包将会自动安装。在其他情况下,请手动安装 CUDA 工具包,请参阅 NVIDIA 官方网站

Pip

需要 Python 3.6 到 3.11,但不推荐使用 3.6。如果您的 Python 低于 3.6.15,将会编译失败。

pip install BabitMF

您还可以在已安装 CUDA 11.8 的 x86_64 arch 的 GPU 主机上安装 GPU 版本:

pip install BabitMF-GPU

对于 C++ 或 Go 开发人员,您需要设置环境变量,以便编译器找到与 BMF 相关的库。您可以执行 bmf_env 来获取路径并在 shell 中执行。

C++ 兼容性注意:不同的编译器对 std::string 类型的定义不同。编译自己的模块时,您可能需要向编译器传递 -D_GLIBCXX_USE_CXX11_ABI=0。是否需要取决于您的编译器使用情况。

支持的操作系统

manylinux x86_64 cpumanylinux x86_64 gpumanylinux i686manylinux aarch64manylinux ppc64lemanylinux s390xmacOS IntelmacOS Apple SiliconmacOS Universal2
CPython 3.6N/AN/A
CPython 3.7N/AN/A
CPython 3.8N/AN/A
CPython 3.9
CPython 3.10

Docker

如果您希望使用 docker 来体验和开始使用 BMF,您可以根据自己的需求编译出所需的版本。我们提供的 docker 镜像基于 ubuntu 20.04,其中包含运行 BMF CPU 和 GPU 的完整环境依赖:Cuda11.8、Pytorch 2.0、TensorRT 8.6.1、CV-CUDA 0.3。 对于 GPU 环境,我们没有安装驱动,因为我们希望尽可能遵循并兼容用户的驱动版本。

在运行之前,请确保您的机器环境中包括 NVIDIA GPU 驱动,并且可以通过 nvidia-smi 正确获取 GPU 硬件信息。

docker pull babitmf/bmf_runtime:latest
docker run --gpus all -e NVIDIA_DRIVER_CAPABILITIES=all -it babitmf/bmf_runtime:latest bash

我们默认提供 BMF CPU 的编译版本。如果您想要 BMF GPU 的编译版本,我们通过环境变量 CMAKE_ARGS 来控制。

export CMAKE_ARGS="-DBMF_ENABLE_CUDA=ON"
./build.sh

预编译二进制文件

发布页面下载,使用 tar 解压,并设置环境变量以查找 BMF。例如:

wget https://github.com/sbravehk/test_pub/releases/download/v0.1.1/bmf-bin-linux-x86_64.tar.gz
tar xvf bmf-bin-linux-x86_64.tar.gz
export C_INCLUDE_PATH=${C_INCLUDE_PATH}:$(pwd)/output/bmf/include
export CPLUS_INCLUDE_PATH=${CPLUS_INCLUDE_PATH}:$(pwd)/output/bmf/include
export LIBRARY_PATH=${LIBRARY_PATH}:$(pwd)/output/bmf/lib
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:$(pwd)/output/bmf/lib

# only set if you want to use BMF in python
export PYTHONPATH=$(pwd)/output/bmf/lib:$(pwd)/output

从源代码构建

BMF 支持在以下三个平台编译和构建:Linux、Windows 和 Mac。您可以根据自己的需要选择您想要使用的方法。

Linux

git clone https://github.com/BabitMF/bmf bmf
cd bmf
./build.sh

在某些特殊情况下,例如,如果用户不想包含 FFmpeg 的依赖项,则可以使用名为 CMAKE_ARGS 的环境变量从源代码构建独立于 FFmpeg 但依赖于 torch 的 BMF:

export CMAKE_ARGS="-DBMF_ENABLE_FFMPEG=OFF -DBMF_ENABLE_TORCH=ON"
./build.sh

还有一些选项需要配置,默认值如下:

option(BMF_ENABLE_BREAKPAD "Enable build with breakpad support" OFF)
option(BMF_ENABLE_CUDA "Enable CUDA support" ON)
option(BMF_ENABLE_TORCH "Enable CUDA support" OFF)
option(BMF_ENABLE_PYTHON "Enable build with python support" ON)
option(BMF_ENABLE_GLOG "Enable build with glog support" OFF)
option(BMF_ENABLE_JNI "Enable build with JNI support" OFF)
option(BMF_ENABLE_FFMPEG "Enable build with ffmpeg support" ON)
option(BMF_ENABLE_MOBILE "Enable build for mobile platform" OFF)
option(BMF_ENABLE_TEST "compile examples and tests" ON)

建议:关于 FFmpeg 兼容性,BMF 目前与 FFmpeg 4.0 - 5.1 版本完全兼容。我们建议使用 4.4 版本。

Windows

BMF 使用 MSVC 工具链作为 Windows 平台上的编译工具。在编译 BMF 之前,您需要完成以下准备工作:

  1. 首先,您需要安装 Visual Studio。我们支持的版本包括 2013、2015、2017、2019 和 2022,并在安装 VS 时安装 Windows SDK。
  2. 安装并配置 msys2 mingw64 环境 https://www.msys2.org/
  3. 完成以上两项后,请以管理员模式打开 x64 Native Tools Command Prompt for VS 20xx,进入 msys2 安装目录,执行 msys2_shell.cmd -mingw64 -use-full-path。您将进入 msys2 shell 命令窗口。
  4. 您需要安装 vcpkg 并执行以下命令来安装一些依赖库:
pacman -Sy yasm automake autoconf git vim openssl-devel zlib-devel
./vcpkg.exe install bzip2:x64-windows zlib:x64-windows liblzma:x64-windows dlfcn-win32:x64-windows
  1. 根据您要编译的产品安装 32 位或 64 位 Python 环境(我们支持 3.7 - 3.10),并配置环境变量。

完成以上步骤后,BMF 编译的准备工作已完成。现在您可以开始编译。编译命令如下:

Build Options:
--msvc msvc compiled version options=[2013, 2015, 2017, 2019, 2022]
bmf_ffmpeg integrates FFmpeg during compilation and compiles built-in Modules
--preset compile preset type options=[x86-Debug, x86-Release, x64-Debug, x64-Release]

假设您的本地环境是 64 位发布版本,您使用的 VS 版本是 2022,编译时需要使用 FFmpeg。编译命令如下:

./build_lite.sh --msvc=2022 bmf_ffmpeg --preset=x64-Release

执行后,build_win_lite 文件夹会生成 BMF.sln 项目文件,通过 Visual Studio 打开即可构建。

Mac OS

Mac OS 端编译时需要注意以下几点:

  1. 安装 FFmpeg 并配置环境变量

  2. 如果您的 Mac 电脑的 CPU 芯片是 M1 或 M2 等 ARM 架构,则兼容的 Python 版本为(3.9 - 3.10)。因为 Mac 上 3.8 以下的 Python 版本是实验性功能,没有太多依赖支持。

  3. 需要安装两个预依赖项:binutils 和 libncurses。在 ARM 架构下,前者可以直接通过 brew install binutils 安装,而后者可能需要您编译 libncursew。 为了在 macOS 上编译 ibncurses,您可以按照以下步骤操作:

    a. 打开终端应用程序并确保已安装 Xcode 命令行工具。如果尚未安装,请在终端中运行以下命令进行安装: xcode-select --install

    b. 下载 ncurses 的源代码。 可以从 ncurses 官网下载最新版本的源码:https://invisible-island.net/ncurses/

    c. 将下载的源码文件解压,进入解压目录:

    tar -xzvf ncurses-x.x.tar.gz # Replace with the downloaded source code file name
    cd ncurses-x.x/ # Enter the decompressed directory
    ./configure --prefix=/usr/local/opt/ncurses
    make
    sudo make install
    

上面的命令将把安装路径配置为 /usr/local/opt/ncurses。您也可以根据需要更改路径。编译安装完成后,可以在指定的安装路径中找到 libncurses 库文件。通过以上步骤,就可以在 mac OS 上成功编译并安装 libncurses。请注意,该流程可能会因版本更新而发生变化。

完成以上两点准备工作后,您可以在 Mac OS 下编译 BMF,使用命令:

./build_osx.sh

安装

您可以将输出目录移至任意位置,然后设置环境变量以查找 BMF 库:

cd ${path_to_bmf}
export C_INCLUDE_PATH=${C_INCLUDE_PATH}:$(pwd)/output/bmf/include
export CPLUS_INCLUDE_PATH=${CPLUS_INCLUDE_PATH}:$(pwd)/output/bmf/include
export LIBRARY_PATH=${LIBRARY_PATH}:$(pwd)/output/bmf/lib
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:$(pwd)/output/bmf/lib

# only set if you want to use BMF in python
export PYTHONPATH=$(pwd)/output/bmf/lib:$(pwd)/output

最后修改 February 19, 2024 : fix baidu statistics (daa0758)