在构建深度学习模型的时候提示Non-OK-status: CudaLaunchKernel(FillPhiloxRandomKernelLaunch, num_blocks, block_size, 0, d.stream(), gen, data, size, dist) status: Internal: out of memory
原因是config.gpu_options.per_process_gpu_memory_fraction = 0.99,比例设置太大了,cuda无法提供,所以报错,改小点就可以了,比如改成0.95
报错 Non-OK-status CudaLaunchKernel
最新推荐文章于 2024-06-19 16:34:13 发布