pytorch3d

mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00

History

Jeremy Reizenstein 174738c33e safer pip install in doc

Summary: Add --no-cache and --no-index to all commands which try to download wheels from S3, to avoid hitting pypi.

Reviewed By: nikhilaravi

Differential Revision: D33507975

fbshipit-source-id: ee796e43cc1864e475cd73c248e9900487012f25

2022-01-21 06:28:32 -08:00

after.sh

Update license for company name

2022-01-04 11:43:38 -08:00

go.sh

Update license for company name

2022-01-04 11:43:38 -08:00

inside.sh

builds for PyTorch 1.10.1

2022-01-10 10:04:02 -08:00

publish.py

Update license for company name

2022-01-04 11:43:38 -08:00

README.md

safer pip install in doc

2022-01-21 06:28:32 -08:00

README.md

Building Linux pip Packages

Make sure this directory is on a filesystem which docker can use - e.g. not NFS. If you are using a local hard drive there is nothing to do here.
You may want to docker pull pytorch/conda-cuda:latest.
Run bash go.sh in this directory. This takes ages and writes packages to inside/output.
You can upload the packages to s3, along with basic html files which enable them to be used, with bash after.sh.

In particular, if you are in a jupyter/colab notebook you can then install using these wheels with the following series of commands.

import sys
import torch
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}"
])
!pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html