该方法安装完 faiss-gpu 之后,不仅会装这个库,还会装很多依赖库。为了防止自己本地一些同名库的版本被修改,建议新建一个虚拟环境来安装。如果本地库版本修改对自己没影响,也可以忽略。
你好,我是悦创。
- 登录网站:https://anaconda.org/ , 然后搜索 faiss-gpu 会进入如下界面。
2. 因为是要在 Windows 上安装,所以选择第二个,点击进入如下界面:
3. 如上图有两条指令,复制其中任意一条命令,然后打开命令行窗口,将命令粘贴过去。这里没有指定对应的 cuda 的版本(默认会装最新的版本),如果需要指定版本的话如下所示:(假设是 11.1 版本)
最好是安装和自己电脑 cuda 对应的版本,以防后续出现其他的问题。
conda install -c conda-forge faiss-gpu cudatoolkit=11.1
输入完命令之后需要等待一段时间,然后弹窗会弹出一个 y/N 的命令,输入 y 然后回车等待就可以了。