423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2026-03-12 23:46:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

pytorch 1.7 support

Browse Source 
Summary: CircleCI build configuration to support pytorch1.7, including binaries with cuda 11.0. Note that the default torch on pip is still on cuda 10.2, so I have left the `main` (non conda build) on cuda 10.2 with the existing driver.

Reviewed By: gkioxari

Differential Revision: D24623523

fbshipit-source-id: 59cfa1be06c16225f0f12ed336c07220e8a9a511
This commit is contained in:
Jeremy Reizenstein
2020-10-29 12:19:24 -07:00
committed by Facebook GitHub Bot
parent fdcf368708
commit 6f4697bc1b
4 changed files with 77 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.circleci/regenerate.py
View File

@@ -19,6 +19,7 @@ CONDA_CUDA_VERSIONS = {
"1.5.0": ["cu92", "cu101", "cu102"],
"1.5.1": ["cu92", "cu101", "cu102"],
"1.6.0": ["cu92", "cu101", "cu102"],
"1.7.0": ["cu101", "cu102", "cu110"],
}
@@ -26,7 +27,7 @@ def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
w = []
for btype in ["conda"]:
for python_version in ["3.6", "3.7", "3.8"]:
for pytorch_version in ["1.4", "1.5.0", "1.5.1", "1.6.0"]:
for pytorch_version in ["1.4", "1.5.0", "1.5.1", "1.6.0", "1.7.0"]:
for cu_version in CONDA_CUDA_VERSIONS[pytorch_version]:
w += workflow_pair(
btype=btype,