diff --git a/conda/faiss-gpu/meta.yaml b/conda/faiss-gpu/meta.yaml index ee9aca6193..e7b839975f 100644 --- a/conda/faiss-gpu/meta.yaml +++ b/conda/faiss-gpu/meta.yaml @@ -89,6 +89,7 @@ outputs: - scipy - pytorch - pytorch-cuda =11.8 + - cudatoolkit =11.8 commands: - python -X faulthandler -m unittest discover -v -s tests/ -p "test_*" - python -X faulthandler -m unittest discover -v -s tests/ -p "torch_*"