“cuDNN不可用”在virtualenv

Question

我想在Theano项目中使用cuDNN。我在Ubuntu上没有sudo权限，并且正在使用virtualenv。

环境variableis设置为cuDNN的存储位置的绝对路径：

LD_LIBRARY_PATH=/home/ME/cudnn:$LD_LIBRARY_PATH

，但我仍然得到错误cuDNN not available。该程序无论如何都在GPU上运行，因此CUDA没有问题。

任何想法？

Answer 1

通过运行sudo ldconfig来更新ldd。您需要在更改LD_LIBRARY_PATH后运行此命令。

ldconfig创建必要的链接并缓存到在命令行，文件/etc/ld.so.conf和受信任的目录（/ lib和/ usr/lib中）。高速缓存由运行时链接程序ld.so或ld-linux.so使用。在确定哪些版本应该链接更新时，ldconfig会检查它遇到的库的头文件和文件名。