提交 4e9de7df 作者: imClumsyPanda

Merge branch 'master' into dev

import torch.cuda
import torch.backends
import torch
def torch_gc(DEVICE):
if torch.cuda.is_available():
......@@ -8,7 +7,6 @@ def torch_gc(DEVICE):
torch.cuda.ipc_collect()
elif torch.backends.mps.is_available():
try:
import torch.mps
torch.mps.empty_cache()
except Exception as e:
print(e)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论