diff --git a/tools/pkg-helpers/pytorch_pkg_helpers/cuda.py b/tools/pkg-helpers/pytorch_pkg_helpers/cuda.py
index a346af8e21..bff75ec1b1 100644
--- a/tools/pkg-helpers/pytorch_pkg_helpers/cuda.py
+++ b/tools/pkg-helpers/pytorch_pkg_helpers/cuda.py
@@ -31,7 +31,7 @@ def get_cuda_variables(
     version_suffix = ""
     pytorch_version_suffix = ""
     wheel_dir = ""
-    if package_type == "wheel" and platform != "darwin" and platform != "linux-aarch64":
+    if package_type == "wheel" and platform != "darwin":
         version_suffix = f"+{gpu_arch_version}"
         pytorch_version_suffix = f"+{gpu_arch_version}"
         wheel_dir = f"{gpu_arch_version}/"