Is there an automatic way to delete all pytorch objects in GPU?

前端 未结 0 656
甜味超标
甜味超标 2021-01-17 05:38

Deleting a specific pytorch object is as simple as

del VGG19 #model
del X_train #tensor

followed by

torch.cuda.empty_cache()
<         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题