Tensorflow可以和Theano一起安裝嗎？

我試圖在Nvidia Tesla K80上安裝Theano上的tensorflow。我正在使用Cuda 7.5並遵循指令here Theano本身運行良好，但只要按照說明或使用pip安裝從源代碼安裝tensorflow，nvidia-smi以及Theano就會停止工作。Tensorflow可以和Theano一起安裝嗎？

更具體地說，nvidia-smi無限期掛起，而Theano只是拒絕在GPU模式下運行。

我也在使用最新版本的cudnn v4。 Tensorflow是否存在導致nvidia-smi掛起並與Theano不兼容的已知問題？

來源

2016-03-28 Sahil

只需安裝TensorFlow不應該在您的計算機上導致其他軟件停止加工。但是，*與Theano同時使用* TensorFlow可能無法正常工作，因爲默認情況下，TensorFlow假定它可以使用GPU中的所有RAM。當你看到這些錯誤時，你是否正在運行TensorFlow程序？ – mrry

嗨。我有一個帶有2個GPU卡的K80 GPU。即使TensorFlow程序在GPU0上運行，在GPU1上運行Theano程序也應該沒問題，對吧？無論tensorflow程序是否在運行，都會發生錯誤。這就是我第一次觀察錯誤的方式，但即使在停止所有程序並重新啓動系統之後，nvidia-smi似乎會永久「損壞」，因爲它無限期地掛起。 'import theano'也一樣。無限期地掛起。 – Sahil

TensorFlow將捕獲所有GPU上的所有RAM，因此您可能在Theano中遇到麻煩。另外，如果你的'nvidia-smi'掛起，那麼GPU相關的任何東西都不會起作用。我遇到了這個問題一次，並通過重新啓動計算機來修復它。 –

TensorFlow挑選所有可用的GPU。因此，如果您在Theano之前啓動它，Theano將不會默認使用任何GPU。如果您先啓動Theano，則TensorFlow會在無法獲取GPU Theano使用時發生段錯誤。

若要解決，使NVIDIA驅動程序只顯示到TensorFlow，你希望他看到這個環境變量的設備：

CUDA_VISIBLE_DEVICES=0 python ...

來源

2016-04-19 13:49:34 nouiz

Tensorflow可以和Theano一起安裝嗎？

回答

相關問題