423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-20 22:30:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

builds for pytorch 1.6

Browse Source 
Summary: Make CI builds with pytorch 1.6.

Reviewed By: gkioxari

Differential Revision: D22790325

fbshipit-source-id: 5b0e075f952de9df2de03b65c16fd69140ab6fdd
This commit is contained in:
Jeremy Reizenstein
2020-07-28 10:38:45 -07:00
committed by Facebook GitHub Bot
parent 9a5341bde3
commit 326f662878
3 changed files with 72 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.circleci/regenerate.py
View File

@@ -17,6 +17,7 @@ CONDA_CUDA_VERSIONS = {
"1.4": ["cu92", "cu100", "cu101"],
"1.5.0": ["cu92", "cu101", "cu102"],
"1.5.1": ["cu92", "cu101", "cu102"],
"1.6.0": ["cu92", "cu101", "cu102"],
}
@@ -24,7 +25,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"]:
for pytorch_version in ["1.4", "1.5.0", "1.5.1", "1.6.0"]:
for cu_version in CONDA_CUDA_VERSIONS[pytorch_version]:
w += workflow_pair(
btype=btype,
@@ -41,7 +42,7 @@ def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
w += workflow_pair(
btype=btype,
python_version=python_version,
pytorch_version="1.5",
pytorch_version="1.6.0",
cu_version=cu_version,
prefix=prefix,
upload=upload,