2016-09-17 63 views
2

我在使用keras時發現了一些奇怪的問題。 我的系統是Ubuntu的14.04/64theano/tensowflow後端的keras錯誤

theano後端錯誤信息:

Using Theano backend. 
Using gpu device 0: Tesla K20Xm (CNMeM is enabled with initial size: 80.0% of memory, cuDNN not available) 
Train on 1000 samples, validate on 100 samples 
Epoch 1/5 
Floating point exception (core dumped) 

tensorflow支持錯誤信息:

F tensorflow/stream_executor/cuda/cuda_driver.cc:316] current context was not created by the StreamExecutor cuda_driver API: 0x2420480; a CUDA runtime call was likely performed without using a StreamExecutor context 

當我使用theano運行CNN和tensowflow運行LSTM這是確定的。 但是當我使用theano運行LSTM或使用tensorflow運行CNN時,錯誤已被佔用。

它通常是一些程序加載80%gpu內存並且從未真正運行。 (CPU結果是正確的)

回答

0

當您有另一個程序在同一時間訪問GPU時,Tensorflow會發生此問題。在你的情況下,顯然Theano是在Tensorflow之前加載的。所以無論在Keras Theano的代碼中的哪個地方都會被意外加載。避免此問題的一個肯定的方法是卸載Theano(從您當前的conda env/virtualenv)。