tensorflow中GPU相关设置解决显存不足

1.显存不足时报错如下：

E tensorflow/stream_executor/cuda/cuda_dnn.cc:359] could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

E tensorflow/stream_executor/cuda/cuda_dnn.cc:326] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM

2018-03-12 10:55:56.078991: F tensorflow/core/kernels/conv_ops.cc:671] Check failed: stream->parent()->GetConvolveAlgorithms(&algorithms) Aborted (core dumped)

Process finished with exit code -1073740791 (0xC0000409)

解决如下：

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=FLAGS.gpu_memory_fraction)
config = tf.ConfigProto(log_device_placement=False,gpu_options=gpu_options)
saver = tf.train.Saver(tf.global_variables())
summary_op = tf.summary.merge_all()
init = tf.initialize_all_variables()
sess = tf.Session(config=config)

tensorflow中GPU相关设置解决显存不足相关推荐

tensorflow中GPU的设置
设置使用哪块GPU: import os os.environ['CUDA_VISIBLE_DEVICES'] = '0' # 使用 GPU 0 os.environ['CUDA_VISIBLE_DE ...
pytorch多gpu DataParallel 及梯度累加解决显存不平衡和显存不足问题
最近在做图像分类实验时,在4个gpu上使用pytorch的DataParallel 函数并行跑程序,批次为16时会报如下所示的错误: RuntimeError: CUDA out of mem ...
3_深度学习中显卡的使用和显存的分配（20181213）
深度学习中显卡的使用和显存的分配(20181213) 1.查看机器上GPU的使用情况 2.tensorflow 中显卡使用和显存分配 1.显卡的使用,单显卡和多显卡调用 1.通过设置环境变量设置禁用 ...
pynvml 查看GPU已使用的显存
Python查看GPU已使用的显存版权声明:随意转载,不用告诉我,但链接到就行. https://blog.csdn.net/u011094454/article/details/80774495 ...
【Ubuntu-Tensorflow】GPU设置及显存资源分配
最近笔者在做GPU显存资源分配的研究,发现在tf中gpu一些实用的方法和接口,共享出来,供大家参考学习,如有问题,欢迎留言讨论. 1.运行程序时,控制台设置GPU运行参数,占用显存全部资源 $ CUD ...
深度学习PyTorch，TensorFlow中GPU利用率较低，CPU利用率很低，且模型训练速度很慢的问题总结与分析
在深度学习模型训练过程中,在服务器端或者本地pc端,输入nvidia-smi来观察显卡的GPU内存占用率(Memory-Usage),显卡的GPU利用率(GPU-util),然后采用top来查看CPU ...
深度学习PyTorch、TensorFlow中GPU利用率与内存占用率很低的问题
上周,在一个使用Pytorch搭建的目标训练项目中,训练时,通过使用命令行执行NVIDIA-SMI(仅支持英伟达显卡)命令发现GPU的利用率基本一直停留在0%,并且显存占用率也较低.CSDN上有一篇分 ...
解决显存不足：Out of video memory trying to allocate a rendering resource
在启动CarlaUE4时,由于我的显卡只有2GB显存,经常出现显存不足问题,导致服务器无法启动,报错为 Out of video memory trying to allocate a renderi ...
如何获取显卡的GPU占用率和显存占用情况
在游戏的性能调优过程中,经常会需要获取CPU占用率.IO.显卡GPU占用率等基础性能数据,下面就简述一下获取nvdia显卡GPU占用率的方法. nvdia 显卡在驱动安装后,可以通过nvapi.dll ...

tensorflow中GPU相关设置解决显存不足

tensorflow中GPU相关设置解决显存不足相关推荐

最新文章

热门文章