# 拉取项目
git clone --recursive https://github.com/pytorch/pytorch
# 切换版本
cd pytorch
git checkout v1.8.2
# 同步子模块代码
git submodule sync
git submodule update --init --recursive
# 获取直接调到GitHub pytorch release 下载稳定版,解压就可以操作,无需同步代码
https://github.com/pytorch/pytorch/releases/download/v1.11.0/pytorch-v1.11.0.tar.gz
apt-get install python3-typing-extensions
apt-get install gcc gfortran
apt-get install ffmpeg
export NO_CUDA=1
export NO_DISTRIBUTED=1
export NO_MKLDNN=1
export MAX_JOBS=4
# 安装命令
python3 setup.py install
# 构建 wheel 包
python3 setup.py bdist_wheel