mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-07-31 10:52:50 +08:00
Summary: This is not actually needed and is causing a conda-forge confusion to do with python_abi - which needs users to have `-c conda-forge` when they install pytorch3d. Reviewed By: patricklabatut Differential Revision: D59587930 fbshipit-source-id: 961ae13a62e1b2b2ce6d8781db38bd97eca69e65
981 B
981 B
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 toinside/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 iopath
!pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html