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.13.0

Browse Source 
Summary: Add builds for current pytorch

Reviewed By: shapovalov

Differential Revision: D41841034

fbshipit-source-id: 4a58515bd5c83b26f14763a3cec9279e905391d2
This commit is contained in:
Jeremy Reizenstein
2022-12-09 07:31:39 -08:00
committed by Facebook GitHub Bot
parent 91b4c04c64
commit 20a774e266
2 changed files with 59 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.circleci/regenerate.py
View File

@@ -28,6 +28,7 @@ CONDA_CUDA_VERSIONS = {
"1.11.0": ["cu102", "cu111", "cu113", "cu115"],
"1.12.0": ["cu102", "cu113", "cu116"],
"1.12.1": ["cu102", "cu113", "cu116"],
"1.13.0": ["cu116", "cu117"],
}
@@ -40,6 +41,8 @@ def conda_docker_image_for_cuda(cuda_version):
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")