GPU的Tensorflow内存问题

时间:2016-04-13 14:27:49

标签: tensorflow

我是Tensorflow的新手。我安装在RHEL7 / CUDA 7.5.18 / libcudnn / 4上。

尝试测试安装时,我继续出现内存/内存分配错误

cannot allocate memory for thread-local data: ABORT.

当我将处理限制为仅CPU(export CUDA_VISIBLE_DEVICES="")时,安装似乎正常工作但是,无论我设置了什么内存限制,当使用GPU执行时,我都会遇到内存错误。

我正在尝试测试MINST演示模型数据集获取这些内存问题(models / image / mnist / convolutional.py)

我谦卑地意识到这很可能是一些天真的配置问题,并且非常感谢任何人愿意提供的帮助。

0 个答案:

没有答案