[NS : Geeen] 恶意占用显存,罪大恶极

importtorchimporttime确保使用GPU4device=torch.device('cuda:4')通过不断增加内存分配来逐步占满GPUtry:whileTrue:#每次分配越来越大的张量,增加内存占用batch_size=300000#逐步增大批量大小#调整张量尺寸x=torch.randn(ba
 
 
Back to Top