Specify CUDA version required

triton-inference-server · Apr 28, 2023 · a3d14f3 · a3d14f3
1 parent 2885c6e
commit a3d14f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py
@@ -218,7 +218,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"