首页
技术日记
编程
旅游
数码
登录
标签
功能解释
清理gpu内存_清理gpu显存
定义一个名为 torch_gc的函数,其目的是释放与PyTorch相关的显存(GPU内存)或其他设备内存。 def torch_gc():if torch.cuda.is_available():
清理
功能解释
编程
admin
10天前
18
0