堆栈内存溢出
cost 105 ms
如何使用CUDNN 5.0构建Tensorflow 1.4？

[英]How to build Tensorflow 1.4 with CUDNN 5.0?

我正在尝试在Centos 7上从具有CUDA 8.0和CUDNN 5.0.5的源安装Tensorflow 1.4，在文档中指出它应与CUDNN 3及更高版本一起使用。 我正在使用带有Bazel 0.7.0和GCC 4.9的Python 3.4.5在虚拟环境中工作。 在配置过程中，我已将CUD ...

预训练张量流模型的烧瓶

[英]flask with pre-trained tensorflow model

有没有办法在烧瓶（0.12.2）应用程序中将预训练的char-lstm张量流与GPU（1.2）模型一起使用？ 通过外壳启动时，模型可以无缝运行。 我正在使用以下代码加载该代码（无需初始化tf变量，占位符或类似代码）： 当试图通过Flask加载相同的预训练模型（带有3个文件： .m ...

conv1d 问题与简单示例

[英]conv1d problems with simple example

关于 conv1d 的两个简单示例：1 维特征：` [1, 0, 2, 3,0, 1, 1] 由 [2, 1, 3] 卷积，输出为 [ 8. 11. 7. 9. 4. ]，这是对的。 但是对于二维特征：“[[1, 0, 2, 3, 0, 1, 1], [1, 0, 2, 3, 0, 1, 1]]”这 ...

关于使用TensorFlow读取tfrecord的问题

[英]Problems about reading tfrecord with tensorflow

现在我的问题是关于读取tfrecords。 例如，假设我有两个图像，并且每个框都有其边界框以包含对象。 数据已成功写入tfrecord文件，现在我的工作是读取它。 当我加载它时，我发现数据不匹配，例如可能是 我尝试使用opencv2绘制它并发现此问题。 我读取tfr ...

在Tesla V100上未启用TF1.4的混合精度

[英]Mixed precision not enabled with TF1.4 on Tesla V100

我有兴趣测试我的神经网络（一个自动编码器，用作发生器+ CNN作为鉴别器），它使用3dconv / deconv层和新的Volta架构，并受益于混合精度训练。 我使用CUDA 9和CudNN 7.0编译了Tensorflow 1.4的最新源代码，并将我的conv / deconv层使用的所有可 ...

为TensorFlow C ++ API的会话选择特定的GPU

[英]Select specific gpu for the session of tensorflow c++ api

我怎样才能让tensorflow使用特定的gpu进行推断？ 部分源代码 到目前为止一切都很好，但是当我执行Run时，tensorflow总是选择相同的gpu，我是否有办法指定要执行的gpu？ 如果您需要完整的源代码，我将它们放在pastebin上 编辑：看起来optio ...

在cuda 9和cudnnv7上安装tensorflow-gpu 1.4

[英]Installation of tensorflow-gpu 1.4 on cuda 9 and cudnnv7

我试图在Windows 10上使用cuda 9和cudnn v7在gtx 1050上安装tensorflow-gpu。我添加了所有环境变量。 但是我收到以下错误：ImportError：找不到'cudart64_80.dll'。 TensorFlow要求将此DLL安装在％PATH％环境变量中 ...

Tensorflow CUDA失败，错误为“无法使流上卷积入队：CUDNN_STATUS_EXECUTION_FAILED”

[英]Tensorflow CUDA fails with error “failed to enqueue convolution on stream: CUDNN_STATUS_EXECUTION_FAILED”

这是我的一些控制台输出。 我不确定真正的问题是什么。 当显示此消息时，我得到一个Windows提示，指出Python.exe已停止工作，原因为ucrtbase.dll，但我尝试对其进行更新，但它仍然发生，因此我认为这是真正问题的结果。 另外，任务栏消息还通知我我的Nvidia Kernal ...

tensorflow-gpu的点安装失败-OSError：原始write（）返回无效长度

[英]Pip install failure for tensorflow-gpu - OSError: raw write() returned invalid length

在Windows 10（Python 3.5）中使用pip安装tensorflow-gpu 1.4.0时，出现以下奇怪错误。 有趣的是，每次我跑步 它会下载该程序包的一个很小但看似随机的部分，然后引发以下异常： 我试图关闭我的防病毒软件，以防万一它被杀了，但是它没有帮助。 ...

Cloud ML Tensorflow和Cudnn版本兼容性

[英]Cloud ML Tensorflow and Cudnn versions compatibility

我想在Cloud ML上使用Tensorflow 1.3（也许是1.4）。 我在Cloud ML的多GPU机器上运行作业 我通过在setup.py中指定tensorflow版本来做到这一点，如下所示： Cloud ML上安装的cudnn库是什么？ 它与tensorflow 1. ...

是否可以同时安装tensorflow的CPU和GPU版本

[英]Is it possible to install the CPU and GPU versions of tensorflow at the same time

我正在使用nvidia-docker从docker容器访问GPU。 但是，并非我们所有的机器都具有GPU，并且当GPU不可用时，我想自动退回到CPU版本。 我是否必须构建单独的docker映像-一个用于CPU且一个用于GPU-还是可以安装tensorflow和tensorflow-gpu ...

有没有办法在GPU上使用tensorflow map_fn？

[英]Is there a way to use tensorflow map_fn on GPU?

我有一个形状为[a，n]的张量A ，我需要对另一个形状为[b，n]的张量B执行op my_op以使所得张量C的形状为[a，b]。 换句话说：对于A中的 每个子张量（A [0]，A 1 ，... A [n]），我需要对B中的 每个子张量执行元素明智的运算。 因此，结果张量将包含以下内容 ...

Keras model.pop() 不工作

[英]Keras model.pop() not working

我正在使用 Python 3.6.2 和 Tensorflow-gpu 1.3.0 运行 Keras 2.0.6。 为了对 Vgg16 模型进行微调，我在手动构建了一个 vgg16 架构并加载了权重后运行了这段代码，但我还没有调用 compile() ： 当我检查 Tensorboard 中的 ...

Tensorflow 1.3和CUDA 8.1

[英]Tensorflow 1.3 and CUDA 8.1

我正在Ubuntu 16.04之上的CUDA 8.0上成功运行Tensorflow 1.3。 安装已按照官方安装说明进行。 1）我想知道Tensorflow是否也与CUDA 9兼容。 支持吗？ 如果是这样，是否会显着提高性能？ 2）如果仅支持CUDA 8：是否支持cuDNN 7 ...

几个时期后的 tensorflow-GPU OOM 问题

[英]tensorflow-GPU OOM issue after several epochs

我用 tensorflow 用 Nvidia Geforce 1060（6G 内存）训练 CNN，但是我得到了一个 OOM 异常。 前两个时期的训练过程很好，但在第三个时期出现了 OOM 异常。 ========================== 2017-10-27 11:47:30.21 ...


排序:
质量:
 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM