如何安装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

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

                                

你可能感兴趣的:(学习工具及问题记录)