pip安装相关依赖
pip3 install numpy
pip3 install future
pip3 install pyyaml
pip3 install requests
pip3 install setuptools
pip3 install six
设置禁用选项
export USE_NCCL=0
export USE_DISTRIBUTED=0
export USE_QNNPACK=0
export USE_PYTORCH_QNNPACK=0
export TORCH_CUDA_ARCH_LIST="5.3;6.2;7.2;"
export PYTORCH_BUILD_NUMBER=1
export USE_CUDNN=0
编译
python3 setup.py bdist_wheel