张量流 GPU 设备"Failed to get device properties"



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

任何想法将不胜感激!

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

最新更新