scripts/check_packages.py

import os

def install_flash_attn():
    print ("installing flash-attn...")
    result = os.system('pip install flash-attn --no-build-isolation')
    print (result)

def install_xformers():
    print ("installing xformers...")
    result = os.system('pip install -U xformers --index-url https://download.pytorch.org/whl/cu129')
    print (result)

def install_kaolin():
    print ("installing kaolin...")
    result = os.system('pip install kaolin==0.18.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.8.0_cu129.html')
    print (result)

def install_nvdiffrast():
    print ("installing nvdiffrast...")
    os.system('mkdir tmp')
    os.system('mkdir tmp/extensions')
    os.system('git clone https://github.com/ZX-Yin/nvdiffrast.git ./tmp/extensions/nvdiffrast')
    result = os.system('pip install ./tmp/extensions/nvdiffrast --no-build-isolation')
    os.system('rm -rf ./tmp')
    print (result)

def install_mipsplatting():
    print ("installing mipsplatting...")
    result = os.system('pip install git+https://github.com/ZX-Yin/mip-splatting-submodules.git --no-build-isolation')
    print (result)

def install_sam2():
    print ("installing SAM 2...")
    result = os.system('pip install git+https://github.com/facebookresearch/sam2.git --no-build-isolation')
    print (result)

def install_utils3d():
    print ("installing utils3d...")
    result = os.system('pip install git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8 --no-build-isolation')
    print (result)