配置Python环境

为了方便进入TVM的Python环境,需要构建一个Python虚拟环境。Python环境推荐使用3.7或3.8.版本,3.9暂时不支持。

  1. 请参见操作系统和软件要求获取Anaconda安装包 。

    1
    wget https://repo.anaconda.com/archive/Anaconda3-2024.10-1-Linux-aarch64.sh
    

  2. 执行安装脚本,并按照提示完成安装以及初始化。

    1
    bash Anaconda3-2024.10-1-Linux-aarch64.sh
    

  3. 重新激活Shell。

    1
    source ~/.bashrc
    

  4. 验证Anaconda是否安装成功。如果Anaconda安装成功,回显信息中会列出已安装的包。

    1
    conda list
    

  5. 使用conda构建一个Python 3.7的虚拟环境并激活。

    1
    2
    conda create -n tvm python=3.7
    conda activate tvm
    

  6. 配置pip源,打开或创建“~/.pip/pip.conf”文件,添加以下内容。

    1
    2
    [global]
    index-url = https://mirrors.huaweicloud.com/repository/pypi/simple/
    

  7. 创建requirements.txt文件,用来指定需要安装的Python依赖。

    numpy==1.19.2
    decorator
    attrs
    typing-extensions
    psutil
    scipy
    tornado
    xgboost==1.5.0
    pybind11
    cython==0.29.32
    pythran
    cloudpickle
    synr==0.6.0
    pytest 
    junitparser 
    onnx

  8. 安装requirements.txt指定的软件包。

    1
    pip install -r requirements.txt