python-3.x - Tensorflow Cuda 错误无法获取设备 0 的设备属性 13
问题描述
我在使用 tensorflow-keras 训练神经网络时遇到问题。我收到此错误:
F tensorflow/stream_executor/lib/statusor.cc:34] 尝试获取值而不是处理错误内部:无法获取设备 0 的设备属性 13:CUDA_ERROR_UNKNOWN:未知错误
我最初收到此错误。然后我参考了failed-to-get-device-attribute-13-for-device-0中的解决方案。我更新了图形驱动程序。这对一些 3-4 运行有效,现在我再次遇到相同的错误。
以下是我的环境的详细信息:
- Python 3.7(蟒蛇)
- 张量流 2.1
- Nvidia GeForce RTX 2060,6GB 显卡
- 视窗 10 版本 1809
解决方案
那么在我的情况下,限制 GPU 内存是有效的。
在代码开头添加以下代码段:
import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(gpus[0], True)
推荐阅读
- php - GA Conversions 将交易链接到客户端 ID theiconic/php-ga-measurement-protocol
- java - 用于当前和启用硒检查的 POP 功能(Java)
- python - 图中的值不清晰可见
- java - 使用 Reactor 和 RabbitMQ 进行反应式编程
- vue.js - VueJs如何将道具类型设置为任何?
- c# - 属性没有隐藏父类属性
- javascript - 在 ES7 中导入样式和图像
- android - NativeScript:从 web 服务获取 json:找不到证书路径的信任锚
- google-sheets - 如何比较 Google 电子表格中一列中的数据
- vba - VBA 几个用户窗体导出到工作表