Skip to content

Commit

Permalink
add scripts for pytorch3d install
Browse files Browse the repository at this point in the history
  • Loading branch information
Kedreamix committed Aug 3, 2024
1 parent c903402 commit a84dc41
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions scripts/install_pytorch3d.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
import sys
import torch
import subprocess

pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
version_str="".join([
f"py3{sys.version_info.minor}_cu",
torch.version.cuda.replace(".",""),
f"_pyt{pyt_version_str}"
])
subprocess.run(["pip", "install", "fvcore", "iopath"])
subprocess.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"])

0 comments on commit a84dc41

Please sign in to comment.