Move pytorch to x86_64 only.

brandon-b-miller · Nov 16, 2022 · c3bdbce · c3bdbce
1 parent 3d91e9f
commit c3bdbce
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -253,8 +253,18 @@ dependencies:
           - pytest-cov
           - pytest-xdist
           - python-snappy>=0.6.0
-          - pytorch<1.12.0
           - s3fs>=2022.3.0
           - scipy
           - tokenizers==0.13.1
           - transformers==4.24.0
+    specific:
+      - output_types: conda
+        matrices:
+          - matrix:
+              arch: x86_64
+            packages:
+              # Currently, CUDA builds of pytorch do not exist for aarch64. We require
+              # version <1.12.0 because newer versions use nvidia::cuda-toolkit.
+              - pytorch<1.12.0
+          - matrix:
+            packages: