「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會檢查它遇到的庫的頭文件和文件名。