diff --git a/cmake/download_onnxruntime.cmake b/cmake/download_onnxruntime.cmake
index 6401c2b..d651e12 100644
--- a/cmake/download_onnxruntime.cmake
+++ b/cmake/download_onnxruntime.cmake
@@ -28,4 +28,5 @@ if(DEFINED TRITON_ONNXRUNTIME_PACKAGE_URL)
 
   endif(NOT DOWNLOAD_RESULT EQUAL 0)
 
-endif(DEFINED TRITON_ONNXRUNTIME_PACKAGE_URL)
\ No newline at end of file
+endif(DEFINED TRITON_ONNXRUNTIME_PACKAGE_URL)
+
diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py
index 82f9554..2a5a4f7 100755
--- a/tools/gen_ort_dockerfile.py
+++ b/tools/gen_ort_dockerfile.py
@@ -256,7 +256,7 @@ def dockerfile_for_linux(output_file):
         if FLAGS.cudnn_home is not None:
             ep_flags += ' --cudnn_home "{}"'.format(FLAGS.cudnn_home)
         elif target_platform() == "igpu":
-            ep_flags += ' --cudnn_home "/usr/lib/aarch64-linux-gnu"'
+            ep_flags += ' --cudnn_home "/usr/include"'
         if FLAGS.ort_tensorrt:
             ep_flags += " --use_tensorrt"
             if FLAGS.ort_version >= "1.12.1":
@@ -275,9 +275,9 @@ def dockerfile_for_linux(output_file):
         ep_flags += (
             " --skip_tests --cmake_extra_defines 'onnxruntime_BUILD_UNIT_TESTS=OFF'"
         )
-        cuda_archs = "53;62;72;87"
+        cuda_archs = "87;101"
     else:
-        cuda_archs = "75;80;86;90"
+        cuda_archs = "75;80;86;89;90;100;120"
 
     df += """
 WORKDIR /workspace/onnxruntime
@@ -472,7 +472,7 @@ def dockerfile_for_windows(output_file):
 WORKDIR /workspace/onnxruntime
 ARG VS_DEVCMD_BAT="\\BuildTools\\VC\\Auxiliary\\Build\\vcvars64.bat"
 RUN powershell Set-Content 'build.bat' -value 'call %VS_DEVCMD_BAT%',(Get-Content 'build.bat')
-RUN build.bat --cmake_generator "Visual Studio 17 2022" --config Release --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=75;80;86;90" --skip_submodule_sync --parallel --build_shared_lib --compile_no_warning_as_error --skip_tests --update --build --build_dir /workspace/build {}
+RUN build.bat --cmake_generator "Visual Studio 17 2022" --config Release --cmake_extra_defines "CMAKE_CUDA_ARCHITECTURES=75;80;86;90;100;120" --skip_submodule_sync --parallel --build_shared_lib --compile_no_warning_as_error --skip_tests --update --build --build_dir /workspace/build {}
 """.format(
         ep_flags
     )