From 94b06bfed30416501c5b02a21824e2fe17169c38 Mon Sep 17 00:00:00 2001 From: Matthijs Douze Date: Wed, 15 May 2024 05:15:50 -0700 Subject: [PATCH] fix install instructions Summary: fix install instruction for GPU + pytorch Reviewed By: mlomeli1 Differential Revision: D57376959 --- INSTALL.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/INSTALL.md b/INSTALL.md index 45e2c9341b..5bd4f6d448 100644 --- a/INSTALL.md +++ b/INSTALL.md @@ -40,11 +40,11 @@ conda install -c pytorch -c nvidia -c rapidsai -c conda-forge faiss-gpu-raft=1.8 ``` In the above commands, pytorch-cuda=11 or pytorch-cuda=12 would select a specific CUDA version, if it’s required. -A combination of versions that installs GPU Faiss with CUDA and Pytorch (as of 2024-03-01): +A combination of versions that installs GPU Faiss with CUDA and Pytorch (as of 2024-05-15): ``` conda create --name faiss_1.8.0 conda activate faiss_1.8.0 -conda install -c pytorch -c nvidia faiss-gpu=1.8.0 pytorch pytorch-cuda numpy +conda install -c pytorch -c nvidia faiss-gpu=1.8.0 pytorch=*=*cuda* pytorch-cuda=11 numpy ``` ## Installing from conda-forge