File tree Expand file tree Collapse file tree 1 file changed +0
-16
lines changed Expand file tree Collapse file tree 1 file changed +0
-16
lines changed Original file line number Diff line number Diff 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 += """
311301RUN ./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"
You can’t perform that action at this time.
0 commit comments