tensorflow-gpu 使用的常见错误

Posted 仰望高端玩家的小清新

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tensorflow-gpu 使用的常见错误相关的知识,希望对你有一定的参考价值。

这篇博客会不定期整理我在 tensorflow 中出现的问题和坑。

1. CUDA_ERROR_OUT_OF_MEMORY:

tensorflow 在执行过程中会默认使用全部的 GPU 内存,给系统保留 200 M,但是在我的系统上会在分配内存时被拒绝导致报错,因此我们可以使用如下语句指定 GPU 内存的分配比例:

# 配置每个 GPU 上占用的内存的比例
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.95)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

可以看看这里的解释:stackoverflow 

还有可能是这个问题:stackoverflow

 

 2. 设置提示信息的等级

# 在 import tensorflow 之前加入

import os
os.environ[TF_CPP_MIN_LOG_LEVEL]=1 # 这是默认的显示等级,显示所有信息

# 2级
import os
os.environ[TF_CPP_MIN_LOG_LEVEL]=2 # 只显示 warning 和 Error

# 3级
import os
os.environ[TF_CPP_MIN_LOG_LEVEL]=3 # 只显示 Error

 

以上是关于tensorflow-gpu 使用的常见错误的主要内容,如果未能解决你的问题,请参考以下文章

二维数组和它的指针

解决报错:在Python中使用property装饰器时,出现错误:TypeError: descriptor ‘setter‘ requires a ‘property‘ object but(代码片

Vue常见问题找不到模块“./App.vue”或其相应的类型声明。ts(2307)

解决报错:在pycharm中使用os模块获取当前进程id,出现错误:AttributeError: module ‘os‘ has no attribute ‘getgid‘(图文并茂!!!)(代码片

矩阵权(Matrix weighted)Bezier三角(曲面)片

我的Android进阶之旅Android Studio 中 使用git提交代码报错:Can‘t commit changes from multiple changelists at once(代码片