faiss安装

按照官网教程,发现安装很慢,并且官网安装时下载的包的版本也不对,conda install -c pytorch很慢,因此选择手动安装包。
1.安装mkl

conda install mkl

2.下载faiss-gpu
首先查看一下自己的cuda版本和python版本,这个一定的要弄对。
然后去 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/ 下载,我下载的是faiss-gpu-1.5.0-py35_cuda10.0_1.tar.bz2
这个压缩包,我的环境是python3.5.6,cuda是10.0。如果清华失效,可以看看这个网站https://conda.anaconda.org/pytorch/linux-64。
3.安装
上传压缩包到linux系统中,手动安装

conda install --use-local faiss-gpu-1.5.0-py35_cuda10.0_1.tar.bz2

测试,如果下面的命令都能执行,说明能够使用。另外,这个faiss也比较耗时,建议模型训练时把评估的间隔调大一些。

python
import faiss
res = faiss.StandardGpuResources()
index = faiss.IndexFlatL2(64)

4.总结
照着别人的做法,把清华源加到conda源中,conda install faiss-gpu,发现下载的版本是错误的,给我下了cuda8的faiss,不能使用,因此自己手动下载了一个最符合的版本,然后本地安装。
按照官方教程,指定了cudatoolkit的版本,但还是给我下载了错误的cuda版本的faiss,无法使用

你可能感兴趣的:(linux,深度学习,python,faiss)