上一篇文章见 天池新闻推荐入门赛——数据分析
前言
需要导入包faiss,先讲解下faiss安装过程。
1.安装mkl
conda install mkl
2.下载faiss-gpu
首先查看一下自己的cuda版本和python版本,这个一定要弄对。
然后去 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/ 下载
这个压缩包,我的环境是python3.6.7,cuda是10.0,下载的是faiss-gpu-1.5.0-py36_cuda10.0_1.tar.bz2。如果清华失效,可以看看这个网站https://conda.anaconda.org/pytorch/linux-64。
3.安装
上传压缩包到linux系统中,手动安装
conda install --use-local faiss-gpu-1.5.0-py36_cuda10.0_1.tar.bz2