安装和使用说明

安装PyTorch 2.4.1

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

1. 下载PyTorch源码和相关第三方依赖，可通过使用git的--recursive选项下载（要求是2.4.1版本）。

   1 2 3 4

   git clone -b v2.4.1 https://github.com/pytorch/pytorch --depth=1 pytorch cd pytorch git submodule sync git submodule update --init --recursive

2. 使用python setup.py install命令安装PyTorch。
   安装前请确认PyTorch对应的Python版本的要求，可在PyTorch官网上查看。。

   1	python setup.py install

3. 验证安装：打开Python终端，输入以下代码：

   1 2	import torch print(torch.__version__)

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

依赖包在setup.py同级目录下的requirement.txt文件中，默认setup.py会直接通过pip install命令安装这些依赖包，如果下载失败，可以手动执行如下依赖包安装命令。

pip install -r requirements.txt

安装Torchvision

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

1. 使用git克隆Torchvision的源代码仓库。

   1	git clone -b v0.20.0 https://github.com/pytorch/vision.git

2. 在“vision”目录中，使用python setup.py develop --cmake命令安装Torchvision。

   1 2	cd vision python setup.py develop --cmake

3. 验证安装：打开Python终端，输入以下代码。

   1 2	import torchvision print(torchvision.__version__)

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

安装Torchvision时需要注意当前Torchvision版本对应的PyTorch版本要求。可在Torchvision官网上查看。

使用PyTorch图算子

1. 从KPCV软件压缩包BoostKit-kp_cv_1.0.0.zip解压得到kpt-2.4.1.patch和kpt.patch，
2. 在PyTorch源码目录下应用kpt-2.4.1.patch文件。

   1 2	git apply kpt-2.4.1.patch python setup.py build

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

   1 2	git apply kpt.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”，提示信息如下所示。

  

  解决方法：在“torch/tools/setup_helpers”路径下的cmake.py和env.py文件中open函数里添加“errors='ignore'”，添加结果如图1和图2所示。

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