scann安装的问题

在CentOS7.4上安装向量搜索scann时,发现出现"Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0" 字样的错误,有2种解决办法:

(1)安装tensorflow-cpu或tensorflow-gpu

根据机器是否支持GPU,选择对应库安装:

pip3 install scann -i https://mirrors.aliyun.com/pypi/simple

pip3 uninstall tensorflow-cpu

pip3 install tensorflow-cpu -i https://mirrors.aliyun.com/pypi/simple

(2)去英伟达官网下载cudatoolkit库

访问: https://developer.nvidia.com/cuda-downloads,去找到对应系统的版本下载安装

另外,注意的是pip安装是自有用户目录,所以别忘记将bin目录加入PATH

export PATH=/$HOME/.local/bin:$PATH

这样就可以按照scann的官方例子运行demo程序,详情demo见(https://github.com/google-research/google-research/blob/master/scann/docs/example.ipynb)

Monthly Archives

Pages

Powered by Movable Type 7.7.2

About this Entry

This page contains a single entry by Cnangel published on June 3, 2021 2:09 PM.

fedora34发布了 was the previous entry in this blog.

离开鹏城的倒计时 is the next entry in this blog.

Find recent content on the main index or look in the archives to find all content.