小编Ani*_*B M的帖子

我正在尝试从 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()

并重新启动没有用的内核

任何帮助，将不胜感激