remove cuda 12 package version name

microsoft · Nov 21, 2023 · 937551d · 937551d
1 parent c145e0c
commit 937551d
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -535,9 +535,6 @@ def finalize_options(self):
                 # Training ROCM package for ADO feeds is called onnxruntime-training-rocm
                 package_name = "onnxruntime-training-rocm"
 
-            if cuda_version and cuda_version.split(".")[0] == "12":
-                package_name = "onnxruntime-training-cuda12"
-
 if package_name == "onnxruntime-tvm":
     packages += ["onnxruntime.providers.tvm"]