Micheal_Yuans的博客

CUDA out of memory: 一种不常见bug的解决方法问题描述解决方案总结 问题描述 博主在跑推荐系统代码的时候出现了OOM(Out Of Memeory)错误，于是把device改为torch.device(“cpu”)，然而又出现了RuntimeError: CUDA out of memory. Tried to allocate 2.00 MiB (GPU 0; 10.91 GiB total capacity; 8.47 GiB already allocated; 1.75 MiB

在处理csv数据的过程中可能会遇到统计重复数据的情况，笔者以自己遇到的问题为例，给出一个可以解决此问题的方法。 进行操作的csv文件中含有1700多条数据，其中局部如下图所示： 目的是计算其中每条数据出现的次数，代码如下： import pandas as pd csv = pd.read_csv('../data/all_CE.csv') counted = csv.groupby(['c...