Skip to content

Commit 153af0e

Browse files
committed
Remove comments
1 parent f019e43 commit 153af0e

File tree

1 file changed

+0
-16
lines changed

1 file changed

+0
-16
lines changed

tools/gen_ort_dockerfile.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -296,16 +296,6 @@ def dockerfile_for_linux(output_file):
296296
""".format(
297297
cuda_archs
298298
)
299-
# if FLAGS.enable_gpu : #and target_platform() != "igpu"
300-
# # For GPU build, include the cudnn_home flag
301-
# df += """
302-
# RUN _CUDNN_VERSION=$(echo $CUDNN_VERSION | cut -d. -f1-2) && ./build.sh ${{COMMON_BUILD_ARGS}} --update --build {} --cudnn_home /usr/local/cudnn-$_CUDNN_VERSION/cuda
303-
# """.format(ep_flags)
304-
# else:
305-
# # For non-GPU
306-
# df += """
307-
# RUN ./build.sh ${{COMMON_BUILD_ARGS}} --update --build {}
308-
# """.format(ep_flags)
309299

310300
df += """
311301
RUN ./build.sh ${{COMMON_BUILD_ARGS}} --update --build {}
@@ -573,14 +563,8 @@ def preprocess_gpu_flags():
573563
FLAGS.tensorrt_home = "/tensorrt"
574564
else:
575565
if "CUDNN_VERSION" in os.environ:
576-
# version = None
577-
# m = re.match(r"([0-9]\.[0-9])\.[0-9]\.[0-9]", os.environ["CUDNN_VERSION"])
578-
# if m:
579-
# version = m.group(1)
580566
if FLAGS.cudnn_home is None:
581-
#FLAGS.cudnn_home = "/usr/local/cudnn-{}/cuda".format(version)
582567
FLAGS.cudnn_home = "/usr"
583-
#FLAGS.cudnn_home = "/usr/lib/x86_64-linux-gnu/"
584568

585569
if FLAGS.cuda_home is None:
586570
FLAGS.cuda_home = "/usr/local/cuda"

0 commit comments

Comments
 (0)