安装faiss-gpu出错的问题
我上网查阅资料,看到有人说需要在后面加GPU或者CPU及。希望我的经历能够对你们有所帮助。
·
最近需要在服务器上的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
最后安装成功了!
希望我的经历能够对你们有所帮助。
更多推荐
所有评论(0)