Apppublic

cnywt/SyncTalk

sourceHugging Facemitupdated 2y agoView on Hugging Face

0likes

install_pytorch3d.py13 linesDownload Raw Back to scripts

1import sys2import torch3import subprocess4 5pyt_version_str=torch.__version__.split("+")[0].replace(".", "")6version_str="".join([7    f"py3{sys.version_info.minor}_cu",8    torch.version.cuda.replace(".",""),9    f"_pyt{pyt_version_str}"10])11subprocess.run(["pip", "install", "fvcore", "iopath"])12subprocess.run(["pip", "install", f"--no-index", f"--no-cache-dir", f"pytorch3d", f"-f", f"https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html"])13