diff --git a/nix-builder/pkgs/python-modules/torch/archs.nix b/nix-builder/pkgs/python-modules/torch/archs.nix
index 3f268db4..43935119 100644
--- a/nix-builder/pkgs/python-modules/torch/archs.nix
+++ b/nix-builder/pkgs/python-modules/torch/archs.nix
@@ -217,5 +217,23 @@
         "9.0"
       ];
     };
+
+    supportedTorchRocmArchs = [
+      # https://github.com/pytorch/pytorch/blob/release/2.11/.ci/docker/almalinux/build.sh
+      "gfx900"
+      "gfx906"
+      "gfx908"
+      "gfx90a"
+      "gfx942"
+      "gfx950"
+      "gfx1030"
+      "gfx1100"
+      "gfx1101"
+      "gfx1102"
+      "gfx1150"
+      "gfx1151"
+      "gfx1200"
+      "gfx1201"
+    ];
   };
 }