python释放cuda缓存库

时间:2022-06-21 19:51:59
【文件属性】:

文件名称:python释放cuda缓存库

文件大小:14KB

文件格式:DLL

更新时间:2022-06-21 19:51:59

python 显存释放

博主自行实现的动态链接库,通过python导入后可以实现释放显存,与ai框架无关。支持pytorch、tensorflow、onnxruntime等cuda运行环境。调用dll.reset_cuda()即可释放显存


网友评论