使用張量流而不使用cuDNN的環境變量不起作用

-1

我不想在沒有sudo權限的計算機上安裝cuDNN。對於我的TensorFlow模型，我不需要cuDNN，因此我使用環境變量TF_USE_CUDNN=0。但是，我得到：使用張量流而不使用cuDNN的環境變量不起作用

ImportError: libcudnn.so.5: cannot open shared object file: No such file or directory

如何在沒有cuDNN的情況下使用TensorFlow？

來源

2017-06-04 user3734670

通過安裝沒有GPU支持的TF？ –

我想使用CUDA但沒有cuDNN，這是可能的根據https://stackoverflow.com/questions/36854940/tensorflow-gpu-is-cudnn-optional-couldnt-open-cuda-library-libcudnn-so – user3734670

提交一個錯誤報告 - 如果沒有遵守TF_USE_CUDNN，沒有意義。 – MaxB

據我所知，從code that reads the environment variable及其調用者可知，TF_USE_CUDNN環境變量僅禁止在運行時使用cuDNN。由於標準的TensorFlow-on-GPU分配鏈接的方式，它仍然需要您安裝cuDNN庫，因爲它將與模塊的其餘部分一起加載。最簡單的解決方法是在您的$HOME/lib目錄中本地安裝cuDNN，並設置LD_LIBRARY_PATH以在那裏查找庫。

來源

2017-06-05 14:25:53 mrry

使用張量流而不使用cuDNN的環境變量不起作用

回答

相關問題