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

Remove the requirement of CUDA's version.txt (#4706)

Browse source 
Sometimes there is a file named "version.txt" in your CUDA installation dir, but sometimes there isn't one. I couldn't figure out it why, but the latest CUDA 11 on our CI build machines doesn't have this file. As the file is not needed for building onnxruntime, so I removed the check.
This commit is contained in:
Changming Sun 2020-08-04 17:03:40 -07:00 committed by GitHub
parent 9d7284fc3b
commit 1e054739b8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tools/ci_build/build.py
View file

@ -946,7 +946,8 @@ def setup_cuda_vars(args):
.format(
cuda_home, cuda_home_valid, cudnn_home, cudnn_home_valid))
if is_windows():
# Our CI build machines already have the env vars setup
if is_windows() and 'AGENT_VERSION' not in os.environ:
# Validate that the cudnn_home is pointing at
# the right level.
if not os.path.exists(os.path.join(cudnn_home, "bin")):