423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

conda pytorch2.0.0 builds (#1480)

Browse Source 
Summary:
Pull Request resolved: https://github.com/facebookresearch/pytorch3d/pull/1480

Nightly builds with PyTorch 2.0.

Reviewed By: shapovalov

Differential Revision: D44135997

fbshipit-source-id: 12b363e2eadbda7a9b6ba9d8db376f41b96d551c
This commit is contained in:
Jeremy Reizenstein 2023-03-21 04:41:37 -07:00 committed by Facebook GitHub Bot
parent 8c8004853a
commit 9437768687
2 changed files with 47 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
.circleci/config.yml
View File

@ -355,6 +355,20 @@ workflows:
name: linux_conda_py38_cu117_pyt1131 name: linux_conda_py38_cu117_pyt1131
python_version: '3.8' python_version: '3.8'
pytorch_version: 1.13.1 pytorch_version: 1.13.1
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda117
context: DOCKERHUB_TOKEN
cu_version: cu117
name: linux_conda_py38_cu117_pyt200
python_version: '3.8'
pytorch_version: 2.0.0
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda118
context: DOCKERHUB_TOKEN
cu_version: cu118
name: linux_conda_py38_cu118_pyt200
python_version: '3.8'
pytorch_version: 2.0.0
- binary_linux_conda: - binary_linux_conda:
context: DOCKERHUB_TOKEN context: DOCKERHUB_TOKEN
cu_version: cu102 cu_version: cu102
@ -530,6 +544,20 @@ workflows:
name: linux_conda_py39_cu117_pyt1131 name: linux_conda_py39_cu117_pyt1131
python_version: '3.9' python_version: '3.9'
pytorch_version: 1.13.1 pytorch_version: 1.13.1
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda117
context: DOCKERHUB_TOKEN
cu_version: cu117
name: linux_conda_py39_cu117_pyt200
python_version: '3.9'
pytorch_version: 2.0.0
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda118
context: DOCKERHUB_TOKEN
cu_version: cu118
name: linux_conda_py39_cu118_pyt200
python_version: '3.9'
pytorch_version: 2.0.0
- binary_linux_conda: - binary_linux_conda:
context: DOCKERHUB_TOKEN context: DOCKERHUB_TOKEN
cu_version: cu102 cu_version: cu102
@ -624,6 +652,20 @@ workflows:
name: linux_conda_py310_cu117_pyt1131 name: linux_conda_py310_cu117_pyt1131
python_version: '3.10' python_version: '3.10'
pytorch_version: 1.13.1 pytorch_version: 1.13.1
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda117
context: DOCKERHUB_TOKEN
cu_version: cu117
name: linux_conda_py310_cu117_pyt200
python_version: '3.10'
pytorch_version: 2.0.0
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda118
context: DOCKERHUB_TOKEN
cu_version: cu118
name: linux_conda_py310_cu118_pyt200
python_version: '3.10'
pytorch_version: 2.0.0
- binary_linux_conda_cuda: - binary_linux_conda_cuda:
name: testrun_conda_cuda_py38_cu102_pyt190 name: testrun_conda_cuda_py38_cu102_pyt190
context: DOCKERHUB_TOKEN context: DOCKERHUB_TOKEN

21
.circleci/regenerate.py
View File

@ -30,32 +30,21 @@ CONDA_CUDA_VERSIONS = {
"1.12.1": ["cu102", "cu113", "cu116"], "1.12.1": ["cu102", "cu113", "cu116"],
"1.13.0": ["cu116", "cu117"], "1.13.0": ["cu116", "cu117"],
"1.13.1": ["cu116", "cu117"], "1.13.1": ["cu116", "cu117"],
"2.0.0": ["cu117", "cu118"],
} }
def conda_docker_image_for_cuda(cuda_version): def conda_docker_image_for_cuda(cuda_version):
if cuda_version in ("cu101", "cu102", "cu111"): if cuda_version in ("cu101", "cu102", "cu111"):
return None return None
if cuda_version == "cu113": if len(cuda_version) != 5:
return "pytorch/conda-builder:cuda113" raise ValueError("Unknown cuda version")
if cuda_version == "cu115": return "pytorch/conda-builder:cuda" + cuda_version[2:]
return "pytorch/conda-builder:cuda115"
if cuda_version == "cu116":
return "pytorch/conda-builder:cuda116"
if cuda_version == "cu117":
return "pytorch/conda-builder:cuda117"
raise ValueError("Unknown cuda version")
def pytorch_versions_for_python(python_version): def pytorch_versions_for_python(python_version):
if python_version in ["3.7", "3.8"]: if python_version in ["3.8", "3.9"]:
return list(CONDA_CUDA_VERSIONS) return list(CONDA_CUDA_VERSIONS)
if python_version == "3.9":
return [
i
for i in CONDA_CUDA_VERSIONS
if version.Version(i) > version.Version("1.7.0")
]
if python_version == "3.10": if python_version == "3.10":
return [ return [
i i