hankerbit的博客

方法一：使用torch.cuda.empty_cache()删除一些不需要的变量代码示例如下：try: output = model(input)except RuntimeError as exception: if "out of memory" in str(exception): print("WARNING: out of memory") if hasattr(torch.cuda, 'empty_cache'): t...

#!/usr/bin/python3# coding: utf-8import torchfrom torchvision import datasets, transformsimport torchvisionfrom tqdm import tqdmdevice_ids = [1, 3, 5,6]BATCH_SIZE = 64# transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize(m