主页
索引
模块索引
搜索页面
手工释放不再使用的 GPU 显存:
import gc del old_model # 这里的old_model是指已经不会再用到的模型 gc.collect() torch.cuda.empty_cache()