该方法安装完faiss-gpu之后,不仅会装这个库,还会装很多依赖库。为了防止自己本地一些同名库的版本被修改,建议新建一个虚拟环境来安装。如果本地库版本修改对自己没影响,也可以忽略。

  1. 登录网站:https://anaconda.org/ , 然后搜索faiss-gpu会进入如下界面。

  1. 因为是要在Windows上安装,所以选择第二个,点击进入如下界面:

  1. 如上图有两条指令,复制其中任意一条命令,然后打开命令行窗口,将命令粘贴过去。这里没有指定对应的cuda的版本(默认会装最新的版本),如果需要指定版本的话如下所示:(假设是11.1版本)

最好是安装和自己电脑cuda对应的版本,以防后续出现其他的问题。

conda install -c conda-forge faiss-gpu cudatoolkit=11.1

4.输入完命令之后需要等待一段时间,然后弹窗会弹出一个y/N的命令,输入y然后回车等待就可以了。

Logo

鸿蒙生态一站式服务平台。

更多推荐