2016-08-06 6 views
0

私はubuntu 16.04でgtx 980tiを実行しています.debファイル経由でcuda 7.5をインストールし、cudnnファイルを/usr/local/cuda-7.5 ...にcupiedしました。 =は/ usr/local/CUDA-7.5 輸出LD_LIBRARY_PATH = $ {CUDA_HOME}/lib64にテンソルフローcudnnの問題(cuda 7.5,367ドライバcudnn 4)SOS

PATH = $ {CUDA_HOME}/binに

輸出CUDA_HOME:

は、私は次のと私のbashrcファイルを編集しました:$ {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. 

誰が何を参照してい私はここで間違っているか、それを修正する方法を知っています。すべての提案は非常に高く評価されています:)

答えて

0

また、パッケージnvidia-cuda-devをインストールする必要があります - これは、libcublas、libcufft、libcurandなどの他のすべての依存関係とそのプレーンな.soシンボリックリンクを引き込みます。

私のUbuntu 16.04システムでは、これらのCUDA 7.5ライブラリはすべて/usr/lib/x86_64-linux-gnuになりましたので、LD_LIBRARY_PATHの設定も必要ありませんでした。

さらに、私はNVIDIA開発者アカウントを通じてlibcudnn5_5.1.3-1+cuda7.5_amd64.deblibcudnn5-dev_5.1.3-1+cuda7.5_amd64.debの2つのパッケージをインストールしました。私は空のLD_LIBRARY_PRELOADで、TensorFlow 0.10構成された環境でipython起動したとき

は今、私は以下を参照してください。

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 

喜び!

関連する問題