擅长:python、mysql、java
<p>我从来没有和Pythorch合作过,但谷歌有几个结果基本上都是一样的。。<code>torch.cuda.empty_cache()</code></p>
<p>{a1}</p>
<p><a href="https://discuss.pytorch.org/t/how-can-we-release-gpu-memory-cache/14530" rel="nofollow noreferrer">https://discuss.pytorch.org/t/how-can-we-release-gpu-memory-cache/14530</a></p>
<p><a href="https://stackoverflow.com/questions/55322434/how-to-clear-cuda-memory-in-pytorch">How to clear Cuda memory in PyTorch</a></p>