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