Cuda out of Memeory Errro when Loading ImageNet dataset

Question

我在加载 ImageNet 数据时遇到问题。我想从图像中提取特征并将它们保存在矩阵中（图像数量，特征大小）。我的 GPU 是带有 24 GB 内存的 Nvidia Quadro RTX 6000。运行我的代码时出现“cuda out of memory”错误。

我试图通过做这些事情来修复它：

1. 降低批量大小（从 256 到 4）和工人数量（从 8 到 2）
2. 设置 max_split_size_mb =2
3. 清除缓存并删除未使用的变量 但他们都没有工作。我怎么解决这个问题？我做错了吗？