Skip to content

Commit 98c37f2

Browse files
committed
Rollback to ORT 1.19.2
1 parent 8c395d8 commit 98c37f2

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -551,8 +551,12 @@ def preprocess_gpu_flags():
551551
FLAGS.tensorrt_home = "/tensorrt"
552552
else:
553553
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)
554558
if FLAGS.cudnn_home is None:
555-
FLAGS.cudnn_home = "/usr"
559+
FLAGS.cudnn_home = "/usr/local/cudnn-{}/cuda".format(version)
556560

557561
if FLAGS.cuda_home is None:
558562
FLAGS.cuda_home = "/usr/local/cuda"

0 commit comments

Comments
 (0)