mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
builds for pytorch 1.6
Summary: Make CI builds with pytorch 1.6. Reviewed By: gkioxari Differential Revision: D22790325 fbshipit-source-id: 5b0e075f952de9df2de03b65c16fd69140ab6fdd
This commit is contained in:
parent
9a5341bde3
commit
326f662878
@ -212,22 +212,22 @@ workflows:
|
|||||||
pytorch_version: "1.4"
|
pytorch_version: "1.4"
|
||||||
cu_version: "cu100"
|
cu_version: "cu100"
|
||||||
- binary_linux_conda_cuda:
|
- binary_linux_conda_cuda:
|
||||||
name: testrun_conda_cuda_py37_cu102_pyt151
|
name: testrun_conda_cuda_py37_cu102_pyt160
|
||||||
python_version: "3.7"
|
python_version: "3.7"
|
||||||
pytorch_version: "1.5.1"
|
pytorch_version: '1.6.0'
|
||||||
cu_version: "cu102"
|
cu_version: "cu102"
|
||||||
- binary_macos_wheel:
|
- binary_macos_wheel:
|
||||||
cu_version: cpu
|
cu_version: cpu
|
||||||
name: macos_wheel_py36_cpu
|
name: macos_wheel_py36_cpu
|
||||||
python_version: '3.6'
|
python_version: '3.6'
|
||||||
pytorch_version: '1.5'
|
pytorch_version: '1.6.0'
|
||||||
- binary_macos_wheel:
|
- binary_macos_wheel:
|
||||||
cu_version: cpu
|
cu_version: cpu
|
||||||
name: macos_wheel_py37_cpu
|
name: macos_wheel_py37_cpu
|
||||||
python_version: '3.7'
|
python_version: '3.7'
|
||||||
pytorch_version: '1.5'
|
pytorch_version: '1.6.0'
|
||||||
- binary_macos_wheel:
|
- binary_macos_wheel:
|
||||||
cu_version: cpu
|
cu_version: cpu
|
||||||
name: macos_wheel_py38_cpu
|
name: macos_wheel_py38_cpu
|
||||||
python_version: '3.8'
|
python_version: '3.8'
|
||||||
pytorch_version: '1.5'
|
pytorch_version: '1.6.0'
|
||||||
|
@ -250,6 +250,21 @@ workflows:
|
|||||||
name: linux_conda_py36_cu102_pyt151
|
name: linux_conda_py36_cu102_pyt151
|
||||||
python_version: '3.6'
|
python_version: '3.6'
|
||||||
pytorch_version: 1.5.1
|
pytorch_version: 1.5.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
cu_version: cu92
|
||||||
|
name: linux_conda_py36_cu92_pyt160
|
||||||
|
python_version: '3.6'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
|
- binary_linux_conda:
|
||||||
|
cu_version: cu101
|
||||||
|
name: linux_conda_py36_cu101_pyt160
|
||||||
|
python_version: '3.6'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
|
- binary_linux_conda:
|
||||||
|
cu_version: cu102
|
||||||
|
name: linux_conda_py36_cu102_pyt160
|
||||||
|
python_version: '3.6'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
- binary_linux_conda:
|
- binary_linux_conda:
|
||||||
cu_version: cu92
|
cu_version: cu92
|
||||||
name: linux_conda_py37_cu92_pyt14
|
name: linux_conda_py37_cu92_pyt14
|
||||||
@ -295,6 +310,21 @@ workflows:
|
|||||||
name: linux_conda_py37_cu102_pyt151
|
name: linux_conda_py37_cu102_pyt151
|
||||||
python_version: '3.7'
|
python_version: '3.7'
|
||||||
pytorch_version: 1.5.1
|
pytorch_version: 1.5.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
cu_version: cu92
|
||||||
|
name: linux_conda_py37_cu92_pyt160
|
||||||
|
python_version: '3.7'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
|
- binary_linux_conda:
|
||||||
|
cu_version: cu101
|
||||||
|
name: linux_conda_py37_cu101_pyt160
|
||||||
|
python_version: '3.7'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
|
- binary_linux_conda:
|
||||||
|
cu_version: cu102
|
||||||
|
name: linux_conda_py37_cu102_pyt160
|
||||||
|
python_version: '3.7'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
- binary_linux_conda:
|
- binary_linux_conda:
|
||||||
cu_version: cu92
|
cu_version: cu92
|
||||||
name: linux_conda_py38_cu92_pyt14
|
name: linux_conda_py38_cu92_pyt14
|
||||||
@ -340,43 +370,58 @@ workflows:
|
|||||||
name: linux_conda_py38_cu102_pyt151
|
name: linux_conda_py38_cu102_pyt151
|
||||||
python_version: '3.8'
|
python_version: '3.8'
|
||||||
pytorch_version: 1.5.1
|
pytorch_version: 1.5.1
|
||||||
- binary_linux_wheel:
|
- binary_linux_conda:
|
||||||
cu_version: cpu
|
cu_version: cu92
|
||||||
name: linux_wheel_py36_cpu_pyt15
|
name: linux_conda_py38_cu92_pyt160
|
||||||
python_version: '3.6'
|
|
||||||
pytorch_version: '1.5'
|
|
||||||
- binary_linux_wheel:
|
|
||||||
cu_version: cpu
|
|
||||||
name: linux_wheel_py37_cpu_pyt15
|
|
||||||
python_version: '3.7'
|
|
||||||
pytorch_version: '1.5'
|
|
||||||
- binary_linux_wheel:
|
|
||||||
cu_version: cpu
|
|
||||||
name: linux_wheel_py38_cpu_pyt15
|
|
||||||
python_version: '3.8'
|
python_version: '3.8'
|
||||||
pytorch_version: '1.5'
|
pytorch_version: 1.6.0
|
||||||
|
- binary_linux_conda:
|
||||||
|
cu_version: cu101
|
||||||
|
name: linux_conda_py38_cu101_pyt160
|
||||||
|
python_version: '3.8'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
|
- binary_linux_conda:
|
||||||
|
cu_version: cu102
|
||||||
|
name: linux_conda_py38_cu102_pyt160
|
||||||
|
python_version: '3.8'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
|
- binary_linux_wheel:
|
||||||
|
cu_version: cpu
|
||||||
|
name: linux_wheel_py36_cpu_pyt160
|
||||||
|
python_version: '3.6'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
|
- binary_linux_wheel:
|
||||||
|
cu_version: cpu
|
||||||
|
name: linux_wheel_py37_cpu_pyt160
|
||||||
|
python_version: '3.7'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
|
- binary_linux_wheel:
|
||||||
|
cu_version: cpu
|
||||||
|
name: linux_wheel_py38_cpu_pyt160
|
||||||
|
python_version: '3.8'
|
||||||
|
pytorch_version: 1.6.0
|
||||||
- binary_linux_conda_cuda:
|
- binary_linux_conda_cuda:
|
||||||
name: testrun_conda_cuda_py37_cu100_pyt14
|
name: testrun_conda_cuda_py37_cu100_pyt14
|
||||||
python_version: "3.7"
|
python_version: "3.7"
|
||||||
pytorch_version: "1.4"
|
pytorch_version: "1.4"
|
||||||
cu_version: "cu100"
|
cu_version: "cu100"
|
||||||
- binary_linux_conda_cuda:
|
- binary_linux_conda_cuda:
|
||||||
name: testrun_conda_cuda_py37_cu102_pyt151
|
name: testrun_conda_cuda_py37_cu102_pyt160
|
||||||
python_version: "3.7"
|
python_version: "3.7"
|
||||||
pytorch_version: "1.5.1"
|
pytorch_version: '1.6.0'
|
||||||
cu_version: "cu102"
|
cu_version: "cu102"
|
||||||
- binary_macos_wheel:
|
- binary_macos_wheel:
|
||||||
cu_version: cpu
|
cu_version: cpu
|
||||||
name: macos_wheel_py36_cpu
|
name: macos_wheel_py36_cpu
|
||||||
python_version: '3.6'
|
python_version: '3.6'
|
||||||
pytorch_version: '1.5'
|
pytorch_version: '1.6.0'
|
||||||
- binary_macos_wheel:
|
- binary_macos_wheel:
|
||||||
cu_version: cpu
|
cu_version: cpu
|
||||||
name: macos_wheel_py37_cpu
|
name: macos_wheel_py37_cpu
|
||||||
python_version: '3.7'
|
python_version: '3.7'
|
||||||
pytorch_version: '1.5'
|
pytorch_version: '1.6.0'
|
||||||
- binary_macos_wheel:
|
- binary_macos_wheel:
|
||||||
cu_version: cpu
|
cu_version: cpu
|
||||||
name: macos_wheel_py38_cpu
|
name: macos_wheel_py38_cpu
|
||||||
python_version: '3.8'
|
python_version: '3.8'
|
||||||
pytorch_version: '1.5'
|
pytorch_version: '1.6.0'
|
||||||
|
@ -17,6 +17,7 @@ CONDA_CUDA_VERSIONS = {
|
|||||||
"1.4": ["cu92", "cu100", "cu101"],
|
"1.4": ["cu92", "cu100", "cu101"],
|
||||||
"1.5.0": ["cu92", "cu101", "cu102"],
|
"1.5.0": ["cu92", "cu101", "cu102"],
|
||||||
"1.5.1": ["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 = []
|
w = []
|
||||||
for btype in ["conda"]:
|
for btype in ["conda"]:
|
||||||
for python_version in ["3.6", "3.7", "3.8"]:
|
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]:
|
for cu_version in CONDA_CUDA_VERSIONS[pytorch_version]:
|
||||||
w += workflow_pair(
|
w += workflow_pair(
|
||||||
btype=btype,
|
btype=btype,
|
||||||
@ -41,7 +42,7 @@ def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
|
|||||||
w += workflow_pair(
|
w += workflow_pair(
|
||||||
btype=btype,
|
btype=btype,
|
||||||
python_version=python_version,
|
python_version=python_version,
|
||||||
pytorch_version="1.5",
|
pytorch_version="1.6.0",
|
||||||
cu_version=cu_version,
|
cu_version=cu_version,
|
||||||
prefix=prefix,
|
prefix=prefix,
|
||||||
upload=upload,
|
upload=upload,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user