builds for PyTorch 2.0.1; drop 1.9

Summary: Drop support for PyTorch 1.9.0 and 1.9.1.

Reviewed By: shapovalov

Differential Revision: D45704329

fbshipit-source-id: c0fe3ecf6a1eb9bcd4163785c0cb4bf4f5060f50
This commit is contained in:
Jeremy Reizenstein
2023-05-10 02:38:47 -07:00
committed by Facebook GitHub Bot
parent c80180c96e
commit 062e6c54ae
3 changed files with 40 additions and 47 deletions

View File

@@ -20,8 +20,6 @@ from packaging import version
# version of pytorch.
# Pytorch 1.4 also supports cuda 10.0 but we no longer build for cuda 10.0 at all.
CONDA_CUDA_VERSIONS = {
"1.9.0": ["cu102", "cu111"],
"1.9.1": ["cu102", "cu111"],
"1.10.0": ["cu102", "cu111", "cu113"],
"1.10.1": ["cu102", "cu111", "cu113"],
"1.10.2": ["cu102", "cu111", "cu113"],
@@ -31,6 +29,7 @@ CONDA_CUDA_VERSIONS = {
"1.13.0": ["cu116", "cu117"],
"1.13.1": ["cu116", "cu117"],
"2.0.0": ["cu117", "cu118"],
"2.0.1": ["cu117", "cu118"],
}