您的当前位置:首页正文

paddle释放显存

2024-11-19 来源:个人技术集锦

参考链接:

import paddle

# required: gpu
paddle.set_device("gpu")
#tensor = paddle.randn([512, 512, 512], "float")
#del tensor
paddle.device.cuda.empty_cache()

显示全文