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