0

ubuntu 16.04でgtx 980tiを実行し、debファイルを介してcuda 7.5をインストールし、cudnnファイルを/ usr/local/cuda-7.5 ...にcupiしました。

次のようにbashrcファイルを編集しました。

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

PATH=${CUDA_HOME}/bin:${PATH} エクスポート 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

さらに、 NVIDIA 開発者アカウントを使用して2 つのパッケージlibcudnn5_5.1.3-1+cuda7.5_amd64.debをインストールしました。libcudnn5-dev_5.1.3-1+cuda7.5_amd64.deb

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 に答える