Update recipe/meta.yaml

Co-authored-by: Mark Harfouche <mark.harfouche@gmail.com>
jeongseok-meta · Nov 4, 2024 · 55f61d4 · 55f61d4
1 parent 620569f
commit 55f61d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recipe/meta.yaml b/recipe/meta.yaml
@@ -349,7 +349,7 @@ outputs:
         - pytorch-cpu                                      # [cuda_compiler_version == "None"]
     requirements:
       run:
-        - pytorch {{ version }}=cuda_{{ blas_impl }}*{{ PKG_BUILDNUM }}   # [megabuild and cuda_compiler_version != "None"]
+        - pytorch {{ version }}=cuda*{{ PKG_BUILDNUM }}                   # [megabuild and cuda_compiler_version != "None"]
         - pytorch {{ version }}=cpu_{{ blas_impl }}*{{ PKG_BUILDNUM }}    # [megabuild and cuda_compiler_version == "None"]
         - {{ pin_subpackage("pytorch", exact=True) }}                     # [not megabuild]
     test: