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 8c395d8 commit 98c37f2Copy full SHA for 98c37f2
tools/gen_ort_dockerfile.py
@@ -551,8 +551,12 @@ def preprocess_gpu_flags():
551
FLAGS.tensorrt_home = "/tensorrt"
552
else:
553
if "CUDNN_VERSION" in os.environ:
554
+ version = None
555
+ m = re.match(r"([0-9]\.[0-9])\.[0-9]\.[0-9]", os.environ["CUDNN_VERSION"])
556
+ if m:
557
+ version = m.group(1)
558
if FLAGS.cudnn_home is None:
- FLAGS.cudnn_home = "/usr"
559
+ FLAGS.cudnn_home = "/usr/local/cudnn-{}/cuda".format(version)
560
561
if FLAGS.cuda_home is None:
562
FLAGS.cuda_home = "/usr/local/cuda"
0 commit comments