Check TensorRT>=8.0.0 version (#6021)
* Check TensorRT>=8.0.0 version * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
1 parent
c1249a47c7
commit
407a905747
2 files changed
+10
-7
No files matched your search
+7
-5
@@ -248,14 +248,16 @@ def check_python(minimum='3.6.2'):
|
||||
check_version(platform.python_version(), minimum, name='Python ', hard=True)
|
||||
|
||||
|
||||
def check_version(current='0.0.0', minimum='0.0.0', name='version ', pinned=False, hard=False):
|
||||
def check_version(current='0.0.0', minimum='0.0.0', name='version ', pinned=False, hard=False, verbose=False):
|
||||
# Check version vs. required version
|
||||
current, minimum = (pkg.parse_version(x) for x in (current, minimum))
|
||||
result = (current == minimum) if pinned else (current >= minimum) # bool
|
||||
if hard: # assert min requirements met
|
||||
assert result, f'{name}{minimum} required by YOLOv5, but {name}{current} is currently installed'
|
||||
else:
|
||||
return result
|
||||
s = f'{name}{minimum} required by YOLOv5, but {name}{current} is currently installed' # string
|
||||
if hard:
|
||||
assert result, s # assert min requirements met
|
||||
if verbose and not result:
|
||||
LOGGER.warning(s)
|
||||
return result
|
||||
|
||||
|
||||
@try_except
|
||||
|
||||
Reference in new issue
Block a user