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

ultralytics 8.0.183 RayTune and yolo checks fixes (#5002)

Browse source 
Co-authored-by: Kapil Raj <103250862+raj-kapil@users.noreply.github.com>
Co-authored-by: Muhammad Rizwan Munawar <62513924+RizwanMunawar@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2023-09-20 16:33:43 +02:00 • committed by GitHub
parent 7f78fad8ba
commit 3223e71fea
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 124 additions and 64 deletions
Download patch file Download diff file Expand all files Collapse all files

10
ultralytics/utils/checks.py
View file

@ -493,9 +493,15 @@ def collect_system_info():
f"{'CPU':<20}{get_cpu_info()}\n"
f"{'CUDA':<20}{torch.version.cuda if torch and torch.cuda.is_available() else None}\n")
for r in parse_requirements():
if (ROOT.parent / 'requirements.txt').exists(): # pip install
requirements = parse_requirements()
else: # git install
from pkg_resources import get_distribution
requirements = get_distribution('ultralytics').requires()
for r in requirements:
current = version(r.name)
is_met = '✅ ' if check_version(current, r.specifier) else '❌ '
is_met = '✅ ' if check_version(current, str(r.specifier)) else '❌ '
LOGGER.info(f'{r.name:<20}{is_met}{current}{r.specifier}')