---
title: 基本安装
description: "PyTorch基本安装的主要步骤包括获取源码与子仓依赖代码、设置环境变量以及编译安装。安装过程中请根据实际情况替换部署路径。"
url: https://www.hikunpeng.com/document/detail/zh/SRA/ecosystemEnable/PyTorch/kunpengpytorch_02_0011.html
sourcePath: /source/zh/SRA/ecosystemEnable/PyTorch/kunpengpytorch_02_0011.html
indexId: c079a756b80ce9e88f76a91cbac31ed192761566f28e39372608de53bed3428066
---
# 基本安装

PyTorch基本安装的主要步骤包括获取源码与子仓依赖代码、设置环境变量以及编译安装。安装过程中请根据实际情况替换部署路径。

1. 配置git网络代理。
  1 2 3 git config --global http.sslVerify false git config --global https.sslVerify false git config --global http.proxy "http://用户名:密码@代理IP地址:代理端口"

2. 获取PyTorch源码。
  1 git clone -b v2.1.2 https://github.com/pytorch/pytorch.git --depth 1

3. 进入“/path/to/pytorch”目录。
  1 cd /path/to/pytorch

4. 获取子仓依赖代码。
  1 2 git submodule sync git submodule update --init --recursive

5. 设置环境变量。
  1 2 3 4 export USE_CUDA=0 export USE_ROCM=0 export USE_FBGEMM=0 export USE_GLOO=0

  USE_CUDA：CUDA是NVIDIA GPU场景下的专有技术，CPU场景设置为0，表示关闭该项。 USE_ROCM：ROCM是AMD GPU场景下的专有技术，CPU场景设置为0，表示关闭该项。 USE_FBGEMM：矩阵运算加速库，只支持x86场景，Arm场景设置为0，表示关闭该项。 USE_GLOO：多个GPU之间和多个节点之间的通信库，CPU场景设置为0，表示关闭该项。

6. 安装Python依赖。
  1 pip install -r requirements.txt

7. 编译安装PyTorch。
  1 python3 setup.py install

8. 安装验证。
  1 pip list | grep torch

  获取PyTorch源码时可能会提示“unexpected disconnect while reading sideband packet”，解决办法请参见获取PyTorch源码失败的解决办法。 获取子仓依赖代码时可能会提示“unexpected disconnect while reading sideband packet fatal”，解决办法请参见获取子仓依赖时失败的解决办法。 编译PyTorch时可能会提示在“/path/to/pytorch/third_party/ios-cmake”目录中无法找到任何CMakeLists.txt,Makefile,setup.py,LICENSE,LICENSE.md,LICENSE.txt文件问题，解决办法请参见编译PyTorch时提示无法找到任何文件问题的解决办法。