We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 05cb262 commit b85e027Copy full SHA for b85e027
tools/gen_ort_dockerfile.py
@@ -579,8 +579,8 @@ def preprocess_gpu_flags():
579
# version = m.group(1)
580
if FLAGS.cudnn_home is None:
581
#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/"
+ FLAGS.cudnn_home = "/usr/include"
+ #FLAGS.cudnn_home = "/usr/lib/x86_64-linux-gnu/"
584
585
if FLAGS.cuda_home is None:
586
FLAGS.cuda_home = "/usr/local/cuda"
0 commit comments