423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity
pytorch3d/packaging/linux_wheels
History
Jeremy Reizenstein d0ca3b9e0c builds for PyTorch 1.9.1 
Summary: Add conda builds for the newly released PyTorch version 1.9.1.

Reviewed By: patricklabatut

Differential Revision: D31140206

fbshipit-source-id: 697549a3ef0db8248f4f9b5c00cf1407296b5022
2021-09-27 04:17:13 -07:00
..
after.sh
License lint codebase
2021-06-22 03:45:27 -07:00
go.sh
License lint codebase
2021-06-22 03:45:27 -07:00
inside.sh
builds for PyTorch 1.9.1
2021-09-27 04:17:13 -07:00
publish.py
License lint codebase
2021-06-22 03:45:27 -07:00
README.md
Build wheels for s3
2021-02-05 05:52:36 -08:00

README.md

Building Linux pip Packages

  1. 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.

  2. You may want to docker pull pytorch/conda-cuda:latest.

  3. Run bash go.sh in this directory. This takes ages and writes packages to inside/output.

  4. 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
version_str="".join([
    f"py3{sys.version_info.minor}_cu",
    torch.version.cuda.replace(".",""),
    f"_pyt{torch.__version__[0:5:2]}"
])
!pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html