ubuntu 16.04 环境安装faiss( CPU和GPU)

本文档基于官方教程,详细介绍了在Ubuntu 16.04上安装Faiss(包括CPU和GPU支持)的步骤,涉及Anaconda2、CUDA 8.0、CuDNN 6.0,以及OpenBLAS的安装与配置。首先安装OpenBLAS,然后下载并编译Faiss,通过测试确保安装成功,最后编译Python接口。
摘要由CSDN通过智能技术生成

主要是基于官网教程,在安装过程中有一些注意事项,在这里做一下说明(强烈推荐参考github给出的官方教程,非常详细,一步一步执行即可,链接https://github.com/facebookresearch/faiss/blob/master/INSTALL.md)

环境:ubunt 16.04,anaconda2(python2.7),gtx1050ti,cuda8.0 ,cudnn 6.0

首先是安装cuda8.0 和cudnn6.0,和其他所有的深度学习框架需要的一样,参考本博客其他教程

1,安装openblas

conda install openblas

2, 下载faiss,修改makefile.inc文件

  1. # 下载FAISS源码  
  2. git clone https://github.com/facebookresearch/faiss.git  
  3. # 进入FAISS源码目录  
  4. cd faiss  
  5. # 根据系统配置编译环境  
  6. cp example_makefiles/makefile.inc.Linux ./makefile.inc  

这里需要进到 faiss目录下的makefile.inc里手动修改,文件中默认使用的是centos,需要把centos部分注释掉,并取消ubuntu16.04的注释

这里给出 修改后的makefile.inc的文件内容:

# Copyright (c) 2015-present, Facebook, Inc.
# All rights reserved.
#
# This source code is licensed under the BSD+Patents license found in the
# LICENSE file in the root directory of this source tree.

# -*- makefile -*-
# tested on CentOS 7, Ubuntu 16 and Ubuntu 14, see below to adjust flags to distribution.

CC=gcc
CXX=g++

CFLAGS=-fPIC -m64 -Wall -g -O3 -mavx -msse4 -mpopcnt -fopenmp -Wno-sign-compare -fopenmp
CXXFLAGS=$(CFLAGS) -std=c++11
LDFLAGS=-g -fPIC  -fopenmp

# common linux flags

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值