diff --git a/recipes/vllm/recipe.yaml b/recipes/vllm/recipe.yaml index 54d32e1de9267..39f5b56013c80 100644 --- 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