最近需要在服务器上的anaconda环境下安装一个faiss库,去网上查了很多资料,看别人是如何下载的,无非就是用pip安装和anaconda安装,但是安装的过程不尽人意,下面说一下我遇到的问题以及如何解决的:

        首先是用pip安装,首先我是直接用如下的简单命令去下载:

pip install faiss

  但是却显示如下错误:

我上网查阅资料,看到有人说需要在后面加GPU或者CPU及

pip install faiss-cpu
pip install faiss-gpu

我发现第一个命令可以正常下载,而第二个命令却依然报错如下:

当我以为已经可以使用时,却发现我要用的函数在faiss-cpu上没有,只有在faiss-gpu上有,所以只能再去下载faiss-gpu,我先将faiss-cpu卸载,然后尝试用conda命令下载faiss-gpu,网上找到各种命令,但是一直显示如下:

最后请教了实验室的学长,学长给解决了,具体如下:

在pycharm专业版的左下角有一个python软件包,打开并搜索faiss-gpu,显示如下:

可以看到,conda一个都没又找到,这可能是因为conda版本不高,但是下面pip和清华源却有三个,如上图所示,而我所使用的服务器CUDA版本为CU12的,用如下命令可以查看cuda版本:

nvidia-smi

显示如下:

可以看到,我所使用的服务器版本为12.2,所以我使用了下面的命令安装:

pip install faiss-gpu-cu12

最后安装成功了!

希望我的经历能够对你们有所帮助。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐