和Drive PX2的編譯沒有什么區別

源碼編譯步驟

注意：當完全按照以下步驟進行編譯時，如果出現錯誤，注意檢查CUDA版本、對應的庫是否存在

# Download PyTorch sources
git clone --recursive --branch <version> http://github.com/pytorch/pytorch
cd pytorch# Set Build Options
export USE_NCCL=0
export USE_DISTRIBUTED=0
export TORCH_CUDA_ARCH_LIST="5.3;6.2;7.2"
export PYTORCH_BUILD_VERSION=<version>  # without the leading 'v', e.g. 1.1.0 for PyTorch v1.1.0
export PYTORCH_BUILD_NUMBER=1# Build wheel for Python 2.7 (to pytorch/dist)
sudo apt-get install python-pip cmake
pip install -U pipsudo pip install -U setuptools
sudo pip install -r requirements.txtpip install scikit-build --user
pip install ninja --user# 或者使用 python setup.py build直接安裝
python setup.py bdist_wheel# Build wheel for Python 3.6 (to pytorch/dist)
sudo apt-get install python3-pip cmake
sudo pip3 install -U setuptoolssudo pip3 install -r requirements.txt
pip3 install scikit-build --user
pip3 install ninja --user# 或者使用 python setup.py build直接安裝
python3 setup.py bdist_wheel