首页 > 解决方案 > TensorFlow GPU 设备“无法获取设备属性”

问题描述

当我session.run()在 Python Tensorflow 中运行我的深度学习模型时,出于某种原因,我收到以下错误:

E tensorflow/core/grappler/clusters/utils.cc:87] Failed to get device properties, error code: 30

Failed to initialize GPU device #0: unknown error

\tensorflow/core/kernels/conv_2d_gpu.h:441] Non‑OK‑status: 

CudaLaunchKernel(ShuffleInTensor3Simple, config.block_count, config.thread_per_block, 0, d.stream(), config.virtual_thread_count, in.data(), combined_dims, out.data()) status: Internal: invalid configuration argument

我正在使用带有集成英特尔显卡和 NVIDIA GeForce GTX 1050 GPU 的 Windows 笔记本电脑。我也在使用 tensorflow-gpu 1.14.0。

有时,如果没有弹出此错误,我的 Spyder IDE 中的内核会重新启动,然后停止工作。如果不是,我从中得到的结果session.run()是完全错误的。

任何想法将不胜感激!

标签: tensorflowgpu

解决方案


我实际上能够通过更新我的 NVIDIA GPU 图形驱动程序来解决它。现在它正在工作!


推荐阅读