Releases: commaai/apex
Releases · commaai/apex
pytorch2.1.0+cu11.8
pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cu118
export MAKEFLAGS="-j$(nproc)"
export TORCH_CUDA_ARCH_LIST="6.1;8.6;8.9"
PYTHONPATH="." python setup.py bdist_wheel --cpp_ext --cuda_ext
pytorch2.0.1+cu11.8
pip install torch==2.0.1 --index-url https://download.pytorch.org/whl/cu118
export MAKEFLAGS="-j$(nproc)"
export TORCH_CUDA_ARCH_LIST="6.1;8.6;8.9"
PYTHONPATH="." python setup.py bdist_wheel --cpp_ext --cuda_extpytorch2.0.0+cu11.8
pip install torch==2.0.0 --index-url https://download.pytorch.org/whl/cu118
export MAKEFLAGS="-j$(nproc)"
export TORCH_CUDA_ARCH_LIST="6.1;8.6;8.9"
PYTHONPATH="." python setup.py bdist_wheel --cpp_ext --cuda_extpytorch1.13.1+cu11.7
pip install https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp38-cp38-linux_x86_64.whl
export MAKEFLAGS="-j$(nproc)"
export TORCH_CUDA_ARCH_LIST="6.1+PTX;8.6+PTX"
PYTHONPATH="." python setup.py bdist_wheel --cpp_ext --cuda_extpytorch1.12.1+cu11.6
pip install https://download.pytorch.org/whl/cu116/torch-1.12.1%2Bcu116-cp38-cp38-linux_x86_64.whl
export MAKEFLAGS="-j$(nproc)"
export TORCH_CUDA_ARCH_LIST="6.1+PTX;8.6+PTX"
PYTHONPATH="." python setup.py bdist_wheel --cpp_ext --cuda_extpytorch1.10.0+cu11.3: minimal bert pipeline parallel test (#1216)
pip install https://download.pytorch.org/whl/cu113/torch-1.10.0%2Bcu113-cp38-cp38-linux_x86_64.whl
export TORCH_CUDA_ARCH_LIST="6.1+PTX;7.0+PTX;7.5+PTX;8.6+PTX"
PYTHONPATH="." python setup.py bdist_wheel --cpp_ext --cuda_ext
pytorch1.10.0+cu11.1
pip install https://download.pytorch.org/whl/cu111/torch-1.10.0%2Bcu111-cp38-cp38-linux_x86_64.whl
export TORCH_CUDA_ARCH_LIST="6.1+PTX;7.0+PTX;7.5+PTX;8.6+PTX"
PYTHONPATH="." python setup.py bdist_wheel --cpp_ext --cuda_ext