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.5.1

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.

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