mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
specify full pytorch version for conda nightly builds.
Summary: Now pytorch 1.5.1 is released, pytorch 1.5 is ambiguous and causes problems. Now have specific builds for pytorch 1.5.0 and 1.5.1. Here we only change the conda builds. Reviewed By: gkioxari Differential Revision: D22196016 fbshipit-source-id: 478327e870f538f54d3480d5a268a1ece5c5c680
This commit is contained in:
parent
2ea6a7d8ad
commit
0baeb05a32
@ -212,9 +212,9 @@ workflows:
|
||||
pytorch_version: "1.4"
|
||||
cu_version: "cu100"
|
||||
- binary_linux_conda_cuda:
|
||||
name: testrun_conda_cuda_py37_cu102_pyt15
|
||||
name: testrun_conda_cuda_py37_cu102_pyt151
|
||||
python_version: "3.7"
|
||||
pytorch_version: "1.5"
|
||||
pytorch_version: "1.5.1"
|
||||
cu_version: "cu102"
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
|
@ -222,19 +222,34 @@ workflows:
|
||||
pytorch_version: '1.4'
|
||||
- binary_linux_conda:
|
||||
cu_version: cu92
|
||||
name: linux_conda_py36_cu92_pyt15
|
||||
name: linux_conda_py36_cu92_pyt150
|
||||
python_version: '3.6'
|
||||
pytorch_version: '1.5'
|
||||
pytorch_version: 1.5.0
|
||||
- binary_linux_conda:
|
||||
cu_version: cu101
|
||||
name: linux_conda_py36_cu101_pyt15
|
||||
name: linux_conda_py36_cu101_pyt150
|
||||
python_version: '3.6'
|
||||
pytorch_version: '1.5'
|
||||
pytorch_version: 1.5.0
|
||||
- binary_linux_conda:
|
||||
cu_version: cu102
|
||||
name: linux_conda_py36_cu102_pyt15
|
||||
name: linux_conda_py36_cu102_pyt150
|
||||
python_version: '3.6'
|
||||
pytorch_version: '1.5'
|
||||
pytorch_version: 1.5.0
|
||||
- binary_linux_conda:
|
||||
cu_version: cu92
|
||||
name: linux_conda_py36_cu92_pyt151
|
||||
python_version: '3.6'
|
||||
pytorch_version: 1.5.1
|
||||
- binary_linux_conda:
|
||||
cu_version: cu101
|
||||
name: linux_conda_py36_cu101_pyt151
|
||||
python_version: '3.6'
|
||||
pytorch_version: 1.5.1
|
||||
- binary_linux_conda:
|
||||
cu_version: cu102
|
||||
name: linux_conda_py36_cu102_pyt151
|
||||
python_version: '3.6'
|
||||
pytorch_version: 1.5.1
|
||||
- binary_linux_conda:
|
||||
cu_version: cu92
|
||||
name: linux_conda_py37_cu92_pyt14
|
||||
@ -252,19 +267,34 @@ workflows:
|
||||
pytorch_version: '1.4'
|
||||
- binary_linux_conda:
|
||||
cu_version: cu92
|
||||
name: linux_conda_py37_cu92_pyt15
|
||||
name: linux_conda_py37_cu92_pyt150
|
||||
python_version: '3.7'
|
||||
pytorch_version: '1.5'
|
||||
pytorch_version: 1.5.0
|
||||
- binary_linux_conda:
|
||||
cu_version: cu101
|
||||
name: linux_conda_py37_cu101_pyt15
|
||||
name: linux_conda_py37_cu101_pyt150
|
||||
python_version: '3.7'
|
||||
pytorch_version: '1.5'
|
||||
pytorch_version: 1.5.0
|
||||
- binary_linux_conda:
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt15
|
||||
name: linux_conda_py37_cu102_pyt150
|
||||
python_version: '3.7'
|
||||
pytorch_version: '1.5'
|
||||
pytorch_version: 1.5.0
|
||||
- binary_linux_conda:
|
||||
cu_version: cu92
|
||||
name: linux_conda_py37_cu92_pyt151
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.5.1
|
||||
- binary_linux_conda:
|
||||
cu_version: cu101
|
||||
name: linux_conda_py37_cu101_pyt151
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.5.1
|
||||
- binary_linux_conda:
|
||||
cu_version: cu102
|
||||
name: linux_conda_py37_cu102_pyt151
|
||||
python_version: '3.7'
|
||||
pytorch_version: 1.5.1
|
||||
- binary_linux_conda:
|
||||
cu_version: cu92
|
||||
name: linux_conda_py38_cu92_pyt14
|
||||
@ -282,19 +312,34 @@ workflows:
|
||||
pytorch_version: '1.4'
|
||||
- binary_linux_conda:
|
||||
cu_version: cu92
|
||||
name: linux_conda_py38_cu92_pyt15
|
||||
name: linux_conda_py38_cu92_pyt150
|
||||
python_version: '3.8'
|
||||
pytorch_version: '1.5'
|
||||
pytorch_version: 1.5.0
|
||||
- binary_linux_conda:
|
||||
cu_version: cu101
|
||||
name: linux_conda_py38_cu101_pyt15
|
||||
name: linux_conda_py38_cu101_pyt150
|
||||
python_version: '3.8'
|
||||
pytorch_version: '1.5'
|
||||
pytorch_version: 1.5.0
|
||||
- binary_linux_conda:
|
||||
cu_version: cu102
|
||||
name: linux_conda_py38_cu102_pyt15
|
||||
name: linux_conda_py38_cu102_pyt150
|
||||
python_version: '3.8'
|
||||
pytorch_version: '1.5'
|
||||
pytorch_version: 1.5.0
|
||||
- binary_linux_conda:
|
||||
cu_version: cu92
|
||||
name: linux_conda_py38_cu92_pyt151
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.5.1
|
||||
- binary_linux_conda:
|
||||
cu_version: cu101
|
||||
name: linux_conda_py38_cu101_pyt151
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.5.1
|
||||
- binary_linux_conda:
|
||||
cu_version: cu102
|
||||
name: linux_conda_py38_cu102_pyt151
|
||||
python_version: '3.8'
|
||||
pytorch_version: 1.5.1
|
||||
- binary_linux_wheel:
|
||||
cu_version: cpu
|
||||
name: linux_wheel_py36_cpu_pyt15
|
||||
@ -316,9 +361,9 @@ workflows:
|
||||
pytorch_version: "1.4"
|
||||
cu_version: "cu100"
|
||||
- binary_linux_conda_cuda:
|
||||
name: testrun_conda_cuda_py37_cu102_pyt15
|
||||
name: testrun_conda_cuda_py37_cu102_pyt151
|
||||
python_version: "3.7"
|
||||
pytorch_version: "1.5"
|
||||
pytorch_version: "1.5.1"
|
||||
cu_version: "cu102"
|
||||
- binary_macos_wheel:
|
||||
cu_version: cpu
|
||||
|
@ -15,7 +15,8 @@ import yaml
|
||||
# version of pytorch.
|
||||
CONDA_CUDA_VERSIONS = {
|
||||
"1.4": ["cu92", "cu100", "cu101"],
|
||||
"1.5": ["cu92", "cu101", "cu102"],
|
||||
"1.5.0": ["cu92", "cu101", "cu102"],
|
||||
"1.5.1": ["cu92", "cu101", "cu102"],
|
||||
}
|
||||
|
||||
|
||||
@ -23,7 +24,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"]:
|
||||
for pytorch_version in ["1.4", "1.5.0", "1.5.1"]:
|
||||
for cu_version in CONDA_CUDA_VERSIONS[pytorch_version]:
|
||||
w += workflow_pair(
|
||||
btype=btype,
|
||||
|
Loading…
x
Reference in New Issue
Block a user