基本安装

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时提示无法找到任何文件问题的解决办法。