" cuDNN不可用"在virtualenv

时间:2017-02-25 21:38:09

标签: ubuntu virtualenv cudnn

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

环境变量设置为cuDNN存储位置的绝对路径:

LD_LIBRARY_PATH=/home/ME/cudnn:$LD_LIBRARY_PATH

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

有什么想法吗?

1 个答案:

答案 0 :(得分:0)

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

  

ldconfig创建必要的链接并缓存到命令行,文件/etc/ld.so.conf和受信任目录(/ lib和/ usr /)中指定的目录中找到的最新共享库。 LIB)。缓存由运行时链接程序ld.so或ld-linux.so使用。在确定哪些版本的链接更新时,ldconfig会检查它遇到的库的头文件名和文件名。