在MUSE Pi Pro上测试SpacemiT Triton

活动比赛Competitions
1

Github仓库：GitHub - spacemit-com/spine-triton

  1. 用vscodium打开文件夹，创建虚拟环境：
python -m venv .venv
source .venv/bin/activate
  1. 环境配置，下载必要的库：
pip install PyYAML sympy torch opencv-python pybind11 --index-url https://git.spacemit.com/api/v4/projects/33/packages/pypi/simple

1056×704 54.2 KB

  1. prebuild whl
pip install triton --index-url https://git.spacemit.com/api/v4/projects/33/packages/pypi/simple

816×304 41.4 KB

  1. Git clone
git clone https://github.com/spacemit-com/spine-triton.git

976×608 71.8 KB

  1. test_smt_mm.py demo
python3 spine-triton/python/examples/test_smt_mm.py

800×480 69.1 KB

PS: 这些库在第二步环境配置时视情况安装：

sudo apt update
sudo apt install libsentencepiece-dev sentencepiece \
    cmake build-essential pkg-config libprotobuf-dev \
    protobuf-compiler libsentencepiece-dev rustup \
    libssl-dev pkg-config llvm llvm-dev \
    libnuma-dev python3-venv python3-full autoconf \
    automake libtool python3-dev python3-pip python3-setuptools \
    libjpeg-dev zlib1g-dev libpng-dev libtiff-dev \
    libfreetype6-dev liblcms2-dev libwebp-dev \
    libopenjp2-7-dev libharfbuzz-dev libfribidi-dev \
    libxcb1-dev python3-virtualenv libsleef3 libsleef-dev libomp-dev libopenmpi-dev \
    libsleef-dev libblas-dev python3-opencv python3-pip
2 个赞