Jeremy Reizenstein c639198c97 builds for PyTorch 1.9
Summary:
Build for pytorch 1.9, and make it the only mac build. Not testing on cuda 11.1, because of annoying failures which are restricted to certain hardware.

Also update cuda driver in CI tests.

Reviewed By: patricklabatut

Differential Revision: D29302314

fbshipit-source-id: 78def378adb9d7aa287abdc5ac0af269e3ba3625
2021-06-22 12:39:44 -07:00
..
2021-06-22 03:45:27 -07:00
2021-06-22 03:45:27 -07:00
2021-06-22 12:39:44 -07:00
2021-06-22 03:45:27 -07:00
2021-02-05 05:52:36 -08:00

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