Try adding a bunch of junk from pytorch recipe

conda-forge · Jan 25, 2025 · 7394748 · 7394748
1 parent 1318567
commit 7394748
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/recipes/vllm/recipe.yaml b/recipes/vllm/recipe.yaml
@@ -35,6 +35,23 @@ requirements:
   - ${{ compiler('c') }}
   - ${{ compiler('cxx') }}
   - ${{ compiler('cuda') }}
+  - cuda-version {{ cuda_compiler_version }}
+  - nccl
+  - cuda-driver-dev                 # [linux]
+  - cuda-cudart-dev
+  - cuda-cupti-dev    # [linux]
+  - cuda-nvrtc-dev
+  - cuda-nvtx-dev
+  - cuda-nvml-dev
+  - cuda-profiler-api
+  - cusparselt     # [linux]
+  - libcublas-dev
+  - libcudss-dev   # [linux]
+  - libcufile-dev  # [linux]
+  - libcufft-dev
+  - libcurand-dev
+  - libcusolver-dev
+  - libcusparse-dev
   run:
   - python
   - psutil