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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM