HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Update torchvision compatibility table for torch 2.6 (#18935)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2025-01-29 12:27:15 +01:00 • committed by GitHub
parent 748c380fc5
commit ed0a7e7dde
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ultralytics/utils/checks.py
View file

@ -433,8 +433,8 @@ def check_torchvision():
The compatibility table is a dictionary where the keys are PyTorch versions and the values are lists of compatible The compatibility table is a dictionary where the keys are PyTorch versions and the values are lists of compatible
Torchvision versions. Torchvision versions.
""" """
# Compatibility table
compatibility_table = { compatibility_table = {
"2.6": ["0.21"],
"2.5": ["0.20"], "2.5": ["0.20"],
"2.4": ["0.19"], "2.4": ["0.19"],
"2.3": ["0.18"], "2.3": ["0.18"],
@ -445,7 +445,7 @@ def check_torchvision():
"1.12": ["0.13"], "1.12": ["0.13"],
} }
# Extract only the major and minor versions # Check major and minor versions
v_torch = ".".join(torch.__version__.split("+")[0].split(".")[:2]) v_torch = ".".join(torch.__version__.split("+")[0].split(".")[:2])
if v_torch in compatibility_table: if v_torch in compatibility_table:
compatible_versions = compatibility_table[v_torch] compatible_versions = compatibility_table[v_torch]