做一个深度学习推荐模型DeepFM的时候出现了这样的错误。完整错误信息:
2021-06-21 14:17:56.689317: F .\tensorflow/core/kernels/random_op_gpu.h:227] Non-OK-status: GpuLaunchKernel(FillPhiloxRandomKernelLaunch, num_blocks, block_size, 0, d.stream(), gen, data, size, dist) status: Internal: invalid device function
怀疑是版本对应问题,python真的好无语,各种版本兼容问题,每天都在搞这些。
参考这个表,更换了一下tensorflow-gpu的版本解决了这个错误,官网更详细链接
DeepFM网络搭建问题请看我的csdn_DeepFM实战篇完全版教程—全资源免费下载