logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

算法模型运行后显存不释放解决办法

第一种方式:利用开子进程的方式进行模型对图像处理,子进程任务执行结束,达到显存释放的目的。第二种方式:torch.cuda.empty_cache()

#pytorch#人工智能#python
到底了