在构建深度学习模型的时候提示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
最新推荐文章于 2025-04-01 11:22:02 发布
本文解决了一个深度学习模型构建过程中的CUDA内存溢出问题。原因是配置的GPU内存使用比例过高,导致CUDA无法分配足够的内存资源,通过将config.gpu_options.per_process_gpu_memory_fraction参数从0.99调整至0.95,成功避免了内存溢出错误。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.6
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
1601





