Skip to content

Commit

Permalink
Remove comments
Browse files Browse the repository at this point in the history
  • Loading branch information
pvijayakrish committed Jan 13, 2025
1 parent f019e43 commit 153af0e
Showing 1 changed file with 0 additions and 16 deletions.
16 changes: 0 additions & 16 deletions tools/gen_ort_dockerfile.py
Original file line number Diff line number Diff line change
Expand Up @@ -296,16 +296,6 @@ def dockerfile_for_linux(output_file):
""".format(
cuda_archs
)
# if FLAGS.enable_gpu : #and target_platform() != "igpu"
# # For GPU build, include the cudnn_home flag
# df += """
# 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
# """.format(ep_flags)
# else:
# # For non-GPU
# df += """
# RUN ./build.sh ${{COMMON_BUILD_ARGS}} --update --build {}
# """.format(ep_flags)

df += """
RUN ./build.sh ${{COMMON_BUILD_ARGS}} --update --build {}
Expand Down Expand Up @@ -573,14 +563,8 @@ def preprocess_gpu_flags():
FLAGS.tensorrt_home = "/tensorrt"
else:
if "CUDNN_VERSION" in os.environ:
# version = None
# m = re.match(r"([0-9]\.[0-9])\.[0-9]\.[0-9]", os.environ["CUDNN_VERSION"])
# if m:
# version = m.group(1)
if FLAGS.cudnn_home is None:
#FLAGS.cudnn_home = "/usr/local/cudnn-{}/cuda".format(version)
FLAGS.cudnn_home = "/usr"
#FLAGS.cudnn_home = "/usr/lib/x86_64-linux-gnu/"

if FLAGS.cuda_home is None:
FLAGS.cuda_home = "/usr/local/cuda"
Expand Down

0 comments on commit 153af0e

Please sign in to comment.