使用tensorflow2.3.1训练模型时,遇到了下面的错误:
StreamExecutor device (0): NVIDIA GeForce MX130, Compute Capability 5.0
2022-02-13 21:03:08.071293: F .\tensorflow/core/kernels/random_op_gpu.h:232] Non-OK-status: GpuLaunchKernel(FillPhiloxRandomKernelLaunch,
num_blocks, block_size, 0, d.stream(), gen, data, size, dist) status: Internal: no kernel image is available for execution on the device
Fatal Python error: Aborted
在CUDA和cuDNN版本正确的前提下出现此错误,是因为tensorflow2.3版本不支持算力5.2以下的英伟达显卡。
因为显卡算力是5.0,所以只能使用tensorflow2.2以下版本
所以只要新创建并安装2.2版本的tensorflow进行训练,就可以解决此问题(或者换电脑)