Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update versions after recent PyTorch releases (#9623)

This commit is contained in:
Carlos Mocholí 2021-09-21 19:49:49 +02:00 committed by GitHub
parent 74c3536e62
commit d022f6ff60
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
requirements/adjust_versions.py
View File

@ -3,16 +3,18 @@ import re
import sys
from typing import Dict, Optional
# IMPORTANT: this list needs to be sorted in reverse
VERSIONS = [
dict(torch="1.10.0", torchvision="0.11.*", torchtext=""), # nightly
dict(torch="1.9.1", torchvision="0.10.1", torchtext="0.10.1"),
dict(torch="1.9.0", torchvision="0.10.0", torchtext="0.10.0"),
dict(torch="1.8.2", torchvision="0.9.1", torchtext="0.9.1"),
dict(torch="1.8.1", torchvision="0.9.1", torchtext="0.9.1"),
dict(torch="1.8.0", torchvision="0.9.0", torchtext="0.9.0"),
dict(torch="1.7.1", torchvision="0.8.2", torchtext="0.8.1"),
dict(torch="1.7.0", torchvision="0.8.1", torchtext="0.8.0"),
dict(torch="1.6.0", torchvision="0.7.0", torchtext="0.7"),
]
VERSIONS.sort(key=lambda v: v["torch"], reverse=True)
def find_latest(ver: str) -> Dict[str, str]: