0

我在 ubuntu 16.04 上运行 gtx 980ti，通过 deb 文件安装 cuda 7.5 并将 cudnn 文件放入 /usr/local/cuda-7.5 ...。

我用以下内容编辑了我的 bashrc 文件：

导出 CUDA_HOME=/usr/local/cuda-7.5 导出 LD_LIBRARY_PATH=${CUDA_HOME}/lib64

PATH=${CUDA_HOME}/bin:${PATH} 导出路径

现在，当我运行模型时，我得到以下信息：

I tensorflow/stream_executor/dso_loader.cc:102] Couldn't open CUDA library libcublas.so. LD_LIBRARY_PATH: 
I tensorflow/stream_executor/cuda/cuda_blas.cc:2289] Unable to load cuBLAS DSO.
I tensorflow/stream_executor/dso_loader.cc:102] Couldn't open CUDA library libcudnn.so. LD_LIBRARY_PATH: 
I tensorflow/stream_executor/cuda/cuda_dnn.cc:2259] Unable to load cuDNN DSO
I tensorflow/stream_executor/dso_loader.cc:102] Couldn't open CUDA library libcufft.so. LD_LIBRARY_PATH: 
I tensorflow/stream_executor/cuda/cuda_fft.cc:343] Unable to load cuFFT DSO.
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:102] Couldn't open CUDA library libcurand.so. LD_LIBRARY_PATH: 
I tensorflow/stream_executor/cuda/cuda_rng.cc:333] Unable to load cuRAND DSO.

有没有人看到我在这里做错了什么，或者知道如何解决它。高度赞赏所有建议:)

4

1 回答 1

1

您还必须安装该软件包nvidia-cuda-dev- 这将引入所有其他依赖项，例如 libcublas、libcufft 和 libcurand，以及它们的普通.so符号链接。

在我的 Ubuntu 16.04 系统上，这些 CUDA 7.5 库全部都包含在内，/usr/lib/x86_64-linux-gnu所以我什至不需要LD_LIBRARY_PATH设置。

libcudnn5_5.1.3-1+cuda7.5_amd64.deb此外，我libcudnn5-dev_5.1.3-1+cuda7.5_amd64.deb通过我的 NVIDIA 开发者帐户安装了这两个软件包。

现在，当我在 TensorFlow 0.10 配置的环境中启动 ipython 时LD_LIBRARY_PRELOAD，我看到：

In [1]: import tensorflow as tf
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:108] successfully opened CUDA library libcurand.so locally

喜悦！

于 2016-09-14T13:33:19.153 回答