CUDA 内存不足。尝试分配 14.00 MiB（GPU 0；4.00 GiB 总容量；2 GiB 已分配；6.20 MiB 可用；PyTorch 总共保留 2GiB）

Question

我正在尝试从 fastai 运行此代码

from fastai.vision.all import *
path = untar_data(URLs.PETS)/'images'

def is_cat(x): return x[0].isupper()
dls = ImageDataLoaders.from_name_func(
    path, get_image_files(path), valid_pct=0.2, seed=42,
    label_func=is_cat, item_tfms=Resize(224), num_workers = 0)

learn = cnn_learner(dls, resnet34, metrics=error_rate)
learn.fine_tune(1)

我收到以下错误

运行时错误：CUDA 内存不足。尝试分配 14.00 MiB（GPU 0；4.00 GiB 总容量；已分配 2.20 GiB；6.20 MiB 可用；PyTorch 总共保留 2.23 GiB）

我也尝试过跑步

import torch
torch.cuda.empty_cache()

并重新启动没有用的内核

任何帮助，将不胜感激

Answer 1

根据此处的文档，默认batch_size使用的是 64 。减少它应该可以解决你的问题。将另一个参数传递给类似的值或任何其他较小的值，直到不抛出错误ImageDataLoaders.from_name_funcImageDataLoaders.from_name_funcbs=32