Specify CUDA version required

triton-inference-server · Apr 10, 2023 · 08dde41 · 08dde41
1 parent 9d81196
commit 08dde41
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py
@@ -221,7 +221,7 @@ def dockerfile_for_linux(output_file):
     if FLAGS.ort_openvino is not None:
         ep_flags += ' --use_openvino CPU_FP32'
 
-    # DLIS-4658: Once Jetson build supports high enough level of CUDA, include compute_90 for Jetson.
+    # DLIS-4658: Once Jetson build supports CUDA 11.8+, include compute_90 for Jetson.
     cuda_archs = "52;60;61;70;75;80;86"
     if target_platform() != 'jetpack':
         cuda_archs += ";90"