---
title: 安装和使用说明
description: "调用PyTorch图算子接口需要预先配置PyTorch和Torchvision环境，并应用相应的补丁文件。"
url: https://www.hikunpeng.com/document/detail/zh/kunpengboostkithistory/251RC1/accel/kunpengaccel_kpcv_06_0033.html
sourcePath: /source/zh/kunpengboostkithistory/251RC1/accel/kunpengaccel_kpcv_06_0033.html
indexId: 9100cfaebcf6698fd7692d4c535e3b33c2d91ea02e4ad3123a269dd634ca07d177
---
# 安装和使用说明

调用PyTorch图算子接口需要预先配置PyTorch和Torchvision环境，并应用相应的补丁文件。

#### 安装conda以及相关依赖

1. 下载安装脚本Miniconda3-py312_25.1.1-2-Linux-aarch64.sh(https://repo.anaconda.com/miniconda/Miniconda3-py312_25.1.1-2-Linux-aarch64.sh)。
2. 执行脚本，根据提示按下回车键或输入“yes”。
  1 2 bash Miniconda3-py312_25.1.1-2-Linux-aarch64.sh source /root/miniconda3/bin/activate

3. 进入conda环境。

```
conda activate base
```


4. 安装依赖。
  1 conda install cmake ninja numpy pyyaml typing_extensions pillow fsspec jinja2 networkx sympy filelock ffmpeg

  conda无法安装完整的GCC、G++，仍需依赖系统环境GCC的库，因此，conda安装的GCC版本和G++的大版本要与系统的一致。例如，gcc -V可以查看版本，若为GCC 12，则执行下面命令。

  1 2 conda install -c conda-forge gcc=12 gxx=12 pip install av==11.0.0 pytest

5. 配置环境变量。
  1 export LD_LIBRARY_PATH=/root/miniconda3/lib:${LD_LIBRARY_PATH}


#### 安装PyTorch 2.4.1

安装PyTorch的依赖项包括Python、pip、CMake以及其他库，安装前确保已安装相关的依赖项。

1. 下载PyTorch源码pytorch-v2.4.1.tar.gz(https://github.com/pytorch/pytorch/releases/download/v2.4.1/pytorch-v2.4.1.tar.gz)。
2. 编译安装。
  安装前请确认PyTorch对应的Python版本的要求，可在PyTorch官网(https://pytorch.org/get-started/locally/)。 1 2 3 tar -xzvf pytorch-v2.4.1.tar.gz && cd pytorch-v2.4.1 && git init && git add . && git commit -m "init" python setup.py build --cmake python setup.py install

3. 打开Python终端，输入以下代码验证PyTorch安装是否成功。注意：不能在pytorch-v2.4.1源码路径下执行，需切换到其他的路径。
  1 2 import torch print(torch.__version__)

  如果输出了PyTorch的版本号，说明PyTorch安装成功。


#### 安装Torchvision

安装前确保已安装完成PyTorch。

1. 下载Torchvision源码v0.20.0.zip(https://github.com/pytorch/vision/archive/refs/tags/v0.20.0.zip)。
2. 编译安装。
  1 2 unzip vision-0.20.0.zip && cd vision-0.20.0 && git init && git add . && git commit -m "init" python setup.py install

3. 打开Python终端，输入以下代码验证Torchvision安装是否成功。注意：不能在vision-0.20.0源码路径下执行，需切换到别的路径。
  1 2 import torchvision print(torchvision.__version__)

  如果输出了Torchvision的版本号，说明Torchvision安装成功。


#### 使用PyTorch图算子

1. 从KPCV软件压缩包BoostKit-kp_cv_***.zip解压得到kpt-2.4.1.patch和kpt.patch和vision-0.20.0.patch。
2. 在PyTorch源码目录下应用kpt-2.4.1.patch文件。resize函数优化需要应用此patch。
  1 2 git apply kpt-2.4.1.patch python setup.py install

3. 在Torchvision源码目录下应用patch文件。

  - normalize和permute的优化需要应用kpt.patch。
    1 2 git apply kpt.patch python setup.py install

  - crop，write_video函数的优化需要应用vision-0.20.0.patch。
    1 2 git apply vision-0.20.0.patch python setup.py install

补丁应用成功后即可使用PyTorch图算子。


#### 故障排除

- 问题1：安装过程中提示immintrin.h头文件找不到。
  解决方法：安装python-devel。

- 问题2：安装过程中提示“'utf-8' codec can't decode byte 0xe0 in position 5560: invalid continuation byte”，提示信息如下所示。
  解决方法：在“pytorch-v2.4.1/tools/setup_helpers”路径下的cmake.py和env.py文件中open函数里添加“errors='ignore'”，添加结果如图1和图2所示。 图1 cmake.py文件open函数添加errors='ignore'

  图2 env.py文件open函数添加errors='ignore'

- 问题3：编译PyTorch时出现“undefined reference to `__cxa_thread_atexit_impl@GLIBC_2.18'”。
  解决方法：

```
export CC=/usr/bin/gcc
export CXX=/usr/bin/g++
```