mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
nightlies with pytorch 1.7.1
Summary: Add nightly linux conda builds with the new pytorch 1.7.1. This supports python 3.9. Reviewed By: nikhilaravi Differential Revision: D25532757 fbshipit-source-id: b734637063e148389951899450566275f3cf5831
This commit is contained in:
parent
1b82388ab8
commit
16a0be790b
@ -308,6 +308,24 @@ workflows:
|
|||||||
name: linux_conda_py36_cu110_pyt170
|
name: linux_conda_py36_cu110_pyt170
|
||||||
python_version: '3.6'
|
python_version: '3.6'
|
||||||
pytorch_version: 1.7.0
|
pytorch_version: 1.7.0
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu101
|
||||||
|
name: linux_conda_py36_cu101_pyt171
|
||||||
|
python_version: '3.6'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu102
|
||||||
|
name: linux_conda_py36_cu102_pyt171
|
||||||
|
python_version: '3.6'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu110
|
||||||
|
name: linux_conda_py36_cu110_pyt171
|
||||||
|
python_version: '3.6'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
- binary_linux_conda:
|
- binary_linux_conda:
|
||||||
context: DOCKERHUB_TOKEN
|
context: DOCKERHUB_TOKEN
|
||||||
cu_version: cu92
|
cu_version: cu92
|
||||||
@ -392,6 +410,24 @@ workflows:
|
|||||||
name: linux_conda_py37_cu110_pyt170
|
name: linux_conda_py37_cu110_pyt170
|
||||||
python_version: '3.7'
|
python_version: '3.7'
|
||||||
pytorch_version: 1.7.0
|
pytorch_version: 1.7.0
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu101
|
||||||
|
name: linux_conda_py37_cu101_pyt171
|
||||||
|
python_version: '3.7'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu102
|
||||||
|
name: linux_conda_py37_cu102_pyt171
|
||||||
|
python_version: '3.7'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu110
|
||||||
|
name: linux_conda_py37_cu110_pyt171
|
||||||
|
python_version: '3.7'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
- binary_linux_conda:
|
- binary_linux_conda:
|
||||||
context: DOCKERHUB_TOKEN
|
context: DOCKERHUB_TOKEN
|
||||||
cu_version: cu92
|
cu_version: cu92
|
||||||
@ -476,6 +512,42 @@ workflows:
|
|||||||
name: linux_conda_py38_cu110_pyt170
|
name: linux_conda_py38_cu110_pyt170
|
||||||
python_version: '3.8'
|
python_version: '3.8'
|
||||||
pytorch_version: 1.7.0
|
pytorch_version: 1.7.0
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu101
|
||||||
|
name: linux_conda_py38_cu101_pyt171
|
||||||
|
python_version: '3.8'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu102
|
||||||
|
name: linux_conda_py38_cu102_pyt171
|
||||||
|
python_version: '3.8'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu110
|
||||||
|
name: linux_conda_py38_cu110_pyt171
|
||||||
|
python_version: '3.8'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu101
|
||||||
|
name: linux_conda_py39_cu101_pyt171
|
||||||
|
python_version: '3.9'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu102
|
||||||
|
name: linux_conda_py39_cu102_pyt171
|
||||||
|
python_version: '3.9'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
|
- binary_linux_conda:
|
||||||
|
context: DOCKERHUB_TOKEN
|
||||||
|
cu_version: cu110
|
||||||
|
name: linux_conda_py39_cu110_pyt171
|
||||||
|
python_version: '3.9'
|
||||||
|
pytorch_version: 1.7.1
|
||||||
- binary_linux_wheel:
|
- binary_linux_wheel:
|
||||||
context: DOCKERHUB_TOKEN
|
context: DOCKERHUB_TOKEN
|
||||||
cu_version: cu101
|
cu_version: cu101
|
||||||
|
@ -20,14 +20,22 @@ CONDA_CUDA_VERSIONS = {
|
|||||||
"1.5.1": ["cu92", "cu101", "cu102"],
|
"1.5.1": ["cu92", "cu101", "cu102"],
|
||||||
"1.6.0": ["cu92", "cu101", "cu102"],
|
"1.6.0": ["cu92", "cu101", "cu102"],
|
||||||
"1.7.0": ["cu101", "cu102", "cu110"],
|
"1.7.0": ["cu101", "cu102", "cu110"],
|
||||||
|
"1.7.1": ["cu101", "cu102", "cu110"],
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def pytorch_versions_for_python(python_version):
|
||||||
|
if python_version in ["3.6", "3.7", "3.8"]:
|
||||||
|
return list(CONDA_CUDA_VERSIONS)
|
||||||
|
pytorch_without_py39 = ["1.4", "1.5.0", "1.5.1", "1.6.0", "1.7.0"]
|
||||||
|
return [i for i in CONDA_CUDA_VERSIONS if i not in pytorch_without_py39]
|
||||||
|
|
||||||
|
|
||||||
def workflows(prefix="", filter_branch=None, upload=False, indentation=6):
|
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", "3.9"]:
|
||||||
for pytorch_version in ["1.4", "1.5.0", "1.5.1", "1.6.0", "1.7.0"]:
|
for pytorch_version in pytorch_versions_for_python(python_version):
|
||||||
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,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user