Tensorflow GPU设备“无法获取设备属性”

时间:2019-12-26 20:40:52

标签: tensorflow gpu

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

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()得到的结果是完全错误的。

任何想法将不胜感激!

1 个答案:

答案 0 :(得分:-1)

我实际上能够通过更新NVIDIA GPU图形驱动程序来解决它。现在可以了!