Skip to content

Commit b85e027

Browse files
committed
Update path
1 parent 05cb262 commit b85e027

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -579,8 +579,8 @@ 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"
583-
FLAGS.cudnn_home = "/usr/lib/x86_64-linux-gnu/"
582+
FLAGS.cudnn_home = "/usr/include"
583+
#FLAGS.cudnn_home = "/usr/lib/x86_64-linux-gnu/"
584584

585585
if FLAGS.cuda_home is None:
586586
FLAGS.cuda_home = "/usr/local/cuda"

0 commit comments

Comments
 (0)