1-删除模型变量
del model_define
2-清空CUDA cache
torch.cuda.empty_cache()
3-步骤2(异步)需要一定时间,设置时延
time.sleep(5)
完整代码如下:
del styler
torch.cuda.empty_cache()
time.sleep(5)
扫码关注
实用AI客栈
获取最新AI资讯与实战案例
小编微信号 : langu86
1-删除模型变量
del model_define
2-清空CUDA cache
torch.cuda.empty_cache()
3-步骤2(异步)需要一定时间,设置时延
time.sleep(5)
完整代码如下:
del styler
torch.cuda.empty_cache()
time.sleep(5)
扫码关注
实用AI客栈
获取最新AI资讯与实战案例
小编微信号 : langu86