Add cuda-version to constrain the related cuda libraries to the same …

…version as the cuda compiler during the build per conda-forge/conda-forge.github.io#1963 fixes conda-forge#112 Signed-off-by: Jonathan Swartz <[email protected]>
swahtz · Jan 16, 2025 · 0286147 · 0286147
1 parent aba3940
commit 0286147
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/recipe/recipe.yaml b/recipe/recipe.yaml
@@ -1,6 +1,6 @@
 context:
   version: 0.20.1
-  build_number: 4
+  build_number: 5
   # see github.com/conda-forge/conda-forge.github.io/issues/1059 for naming discussion
   # torchvision requires that CUDA major and minor versions match with pytorch
   # https://github.com/pytorch/vision/blob/fa99a5360fbcd1683311d57a76fcc0e7323a4c1e/torchvision/extension.py#L79C1-L85C1
@@ -133,6 +133,7 @@ outputs:
         - setuptools
         - if: cuda_compiler_version != "None"
           then:
+            - cuda-version {{ cuda_compiler_version }}
             - cudnn
             - libcublas-dev
             - libcusolver-dev