Skip to content

Commit f019e43

Browse files
committed
Update CUDNN home
1 parent b85e027 commit f019e43

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -579,7 +579,7 @@ def preprocess_gpu_flags():
579579
# version = m.group(1)
580580
if FLAGS.cudnn_home is None:
581581
#FLAGS.cudnn_home = "/usr/local/cudnn-{}/cuda".format(version)
582-
FLAGS.cudnn_home = "/usr/include"
582+
FLAGS.cudnn_home = "/usr"
583583
#FLAGS.cudnn_home = "/usr/lib/x86_64-linux-gnu/"
584584

585585
if FLAGS.cuda_home is None:

0 commit comments

Comments
 (0)