[英]Pytorch CUDA out of memory Vram is not freed after calaculations
我在 Pytorch 中按 kfold 交叉验证训练和评估变压器和 RNN，并且总是在一些模型计算之后（整个模型已经用一些超参数训练和评估）我在计算下一个模型（使用新的超参数）时得到一个 cuda 内存不足错误. 如果我开始运行包含不同模型的完全不同的 iypython 笔记本，甚至会发生这种情况。
不知何故，Vram 没有被释放。 我已经试过了
torch.cuda.empty_cache()
在并行过程中评估 PyTorch 模型上的数据后，GPU 内存不会被释放
[英]GPU memory doesn't get freed up after evaluating data on PyTorch model in parallel process
在 Dask-cudf 处理“std::bad_alloc:out_of_memory:CUDA 错误”
[英]Handle "std::bad_alloc: out_of_memory: CUDA error" at Dask-cudf
声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.