如何安装faiss-gpu ubutnu18 cuda10.1

使用Anaconda安装faiss是最方便快速的方式，facebook会及时推出faiss的新版本conda安装包，在conda安装时会自行安装所需的libgcc, mkl, numpy模块。
faiss的cpu版本目前仅支持Linux和MacOS操作系统，gpu版本提供可在Linux操作系统下用CUDA8.0/CUDA9.0/CUDA9.1编译的版本。

Faiss: 使用conda安装faiss-cpu库

RUN conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ \
    && conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ \
    && conda config --set show_channel_urls yes \
    && conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ \
    && conda install faiss-cpu -c pytorch

 
   
   
   
   

 
   
   
   
   

    
    
    
    
1
    
    
    
    
2
    
    
    
    
3
    
    
    
    
4
    
    
    
    
5
   
   
   
   

conda改源之路

conda config --show-sources            #查看当前使用源
conda config --remove channels 源名称或链接                 #删除指定源
conda config --add channels 源名称或链接                #添加指定源

 
   
   
   
   

 
   
   
   
   

    
    
    
    
1
    
    
    
    
2
    
    
    
    
3
   
   
   
   

需要为conda添加下面两个清华源分支：

https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

 
   
   
   
   

 
   
   
   
   

    
    
    
    
1
    
    
    
    
2
   
   
   
   

命令如下：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --set show_channel_urls yes
conda config --show-sources

 
   
   
   
   

 
   
   
   
   

    
    
    
    
1
    
    
    
    
2
    
    
    
    
3
    
    
    
    
4
   
   
   
   

这里！！请把配置文件~/.condarc中defaults去掉！！！

vim ~/.condarc

 
   
   
   
   

 
   
   
   
   

    
    
    
    
1
   
   
   
   

再运行：

conda install faiss-cpu -c pytorch 

 
   
   
   
   

 
   
   
   
   

    
    
    
    
1
   
   
   
   

不然，此时若直接将命令复制至终端运行，用的依然是默认的URL,因为conda install的-c参数表示指定对应的channel安装，速度非常慢。报错如下：

CondaHTTPError: HTTP 000 CONNECTION FAILED for url //conda.anaconda.org/pytorch/linux-64/faiss-cpu-1.6.3-py37h6bb024c_0.tar.bz2>

 
   
   
   
   

 
   
   
   
   

    
    
    
    
1
   
   
   
   

删除安装命令最后的 -c pytorch，才会采用清华源安装。

conda install faiss-cpu

 
   
   
   
   

 
   
   
   
   

    
    
    
    
1
   
   
   
   

总之，我哭了。希望你看到这个文章，可以减少你心态崩溃的时间。祝好。。