Skip to content

Commit

Permalink
Update INSTALL.md (facebookresearch#2853)
Browse files Browse the repository at this point in the history
Summary:
Update the INSTALL to get a recipe "that works" to install GPU faiss (which is not obvious).

Pull Request resolved: facebookresearch#2853

Reviewed By: algoriddle

Differential Revision: D45652590

Pulled By: mdouze

fbshipit-source-id: fededbdfda33e7a44d279e6bf16da316cff6fcb0
  • Loading branch information
mdouze authored and facebook-github-bot committed May 15, 2023
1 parent f809cf0 commit 54d331e
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions INSTALL.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,13 +31,12 @@ $ conda install -c pytorch/label/nightly faiss-cpu
$ conda install -c pytorch/label/nightly faiss-gpu
```

A combination of versions that works with Pytorch (as of 2022-11-23):
A combination of versions that installs GPU Faiss with CUDA 11.4 and Pytorch (as of 2023-05-08):
```
conda create -n faiss_1.7.3 python=3.8
conda activate faiss_1.7.3
conda install pytorch==1.11.0 cudatoolkit=11.3 -c pytorch
conda install numpy
conda install -c pytorch faiss-gpu=1.7.3 cudatoolkit=11.3
conda create --name faiss_1.7.4 python=3.10
conda activate faiss_1.7.4
conda install faiss-gpu=1.7.4 -c pytorch -c nvidia
conda install faiss-gpu pytorch pytorch-cuda -c pytorch -c nvidia
conda install -c conda-forge notebook
conda install -y matplotlib
```
Expand Down

0 comments on commit 54d331e

Please sign in to comment.