在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)