无法在Ubuntu(Titan-X pascal)上使用GPU处理器用于tensorflow LinearRegressor

时间:2017-01-30 22:52:04

标签: ubuntu tensorflow gpu

我有一个(CUDA 8) - 在Ubuntu 16.04上使用Titan-X Pascal的3个单位,张量为v0.12。 (Nvidia驱动程序版本:370.28)

我无法运行任何contrib.learn模块(LinearRegressor)来使用处理器(GPU-Util在nvidia-smi中始终显示为0%)

我知道我的设置是正确的,因为cifar10 multi-gpu示例(这里:https://github.com/tensorflow/models/blob/master/tutorials/image/cifar10/cifar10_multi_gpu_train.py)使用所有GPU都可以正常工作。

我还在这里提交了有关tensorflow的问题:https://github.com/tensorflow/tensorflow/issues/6132

注意: - 我已将run_config更改为set_soft_placement = True - 我尝试将特定的GPU分别添加到model_fn以及feed_fn,但没有运气。

还有其他人遇到contrib.learn包的同样问题吗?运气好吗?

0 个答案:

没有答案