Pytorch源码编译安装

Pytorch源码编译安装

安装依赖

pip install astunparse numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses

下载Pytorch源码

git clone --branch v1.10.1 --recursive https://github.com/pytorch/pytorch
cd pytorch

note:此时会报错

此时会报错pytorch/third_party/breakpad/src/third_party/lss下载不到

解决途径：
手动下载，lss其实就是linux-syscall-support(lss)
https://chromium.googlesource.com/linux-syscall-support/
下载对应的版本的uuid就行
放在pytorch/third_party/breakpad/src/third_party/，并改名字为lss

编译安装

python setup.py develop

---

安装Torchvision

Download

git clone --recursive --branch v0.11.1 https://github.com/pytorch/vision.git

cd vision
python setup.py develop