Question

向GPU发送单个变量后，我似乎无法清除GPU内存。

import torch
tm = torch.Tensor([1,2]).to("cuda")
!nvidia-smi

|===============================+======================+======================|
|   0  GeForce RTX 208...  On   | 00000000:3D:00.0 Off |                  N/A |
|  0%   37C    P2    52W / 250W |    730MiB / 10989MiB |      0%      Default

所以我用730MiB ... 现在，无论我如何尝试，都无法使730MiB变为零：

del tm                                                                                                                                                                                                 
torch.cuda.empty_cache()                                                                                                                                                                               
import sys;sys.modules[__name__].__dict__.clear()                                                                                                                                                      
%reset                                                                                                                                                                                                 
Once deleted, variables cannot be recovered. Proceed (y/[n])? y
!nvidia-smi
|   0  GeForce RTX 208...  On   | 00000000:3D:00.0 Off |                  N/A |
|  0%   35C    P8     1W / 250W |    728MiB / 10989MiB |      0%      Default |

我很高兴听到任何建议，谢谢

Answer 1

好的，不可能，此内存是割炬驱动程序，无法释放。我已经在pytorch GitHub-https://github.com/pytorch/pytorch/issues/37664

中打开了一张票

完全清晰的GPU内存

1 个答案: