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一起安装吗？

回答

相关问题