423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-23 06:12:48 +08:00
Code Issues Packages Projects Releases Wiki Activity
pytorch3d/packaging/linux_wheels
History
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
..
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
2021-06-22 12:39:44 -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