Chen
Results
1
comments of
Chen
import torch torch.cuda.empty_cache() # 在每次推理结束后，释放显存 希望可以帮助到你。