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

PyTorch 1.10 + CUDA 11.1 builds

Browse Source 
Summary: Although the PyTorch website, which describes the current version 1.10, suggests CUDA 10.2 and 11.3 are supported, it would appear that we need to include builds for CUDA 11.1 to avoid surprises. This is because these builds are on anaconda, and this combination is used on Google Colab.

Reviewed By: nikhilaravi

Differential Revision: D33063932

fbshipit-source-id: 1b22d1f06e22bd18fb53ceecb58e78ac6a5d1693
This commit is contained in:
Jeremy Reizenstein
2021-12-13 10:10:02 -08:00
committed by Facebook GitHub Bot
parent 1edc624d82
commit d049cd2e01
2 changed files with 25 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.circleci/regenerate.py
View File

@@ -26,7 +26,7 @@ CONDA_CUDA_VERSIONS = {
"1.8.1": ["cu101", "cu102", "cu111"],
"1.9.0": ["cu102", "cu111"],
"1.9.1": ["cu102", "cu111"],
"1.10.0": ["cu102", "cu113"],
"1.10.0": ["cu102", "cu111", "cu113"],
}