saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-28 22:56:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

do not post cuda version mismatch warning if cannot find local cudart version (#8924)

Browse source 
Co-authored-by: liqun <liqun@OrtTrainingDev4.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
This commit is contained in:
liqun Fu 2021-09-01 17:11:54 -07:00 committed by GitHub
parent f126a12699
commit 757e9e6df7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/onnxruntime_validation.py
View file

@ -117,7 +117,7 @@ def validate_build_package_info():
# collection cuda library info from current environment.
from onnxruntime.capi.onnxruntime_collect_build_info import find_cudart_versions
local_cudart_versions = find_cudart_versions(build_env=False, build_cuda_version=cuda_version)
if cudart_version and cudart_version not in local_cudart_versions:
if cudart_version and local_cudart_versions and cudart_version not in local_cudart_versions:
print_build_package_info()
warnings.warn('WARNING: failed to find cudart version that matches onnxruntime build info')
warnings.warn('WARNING: found cudart versions: %s' % local_cudart_versions)