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