求助tensorflow大佬

lemeaco · 发表于 2021-11-10 21:44:52

如图，调用tf.data.Dataset.from_tensor_slices()接口把pandas读出来的文件转成tf格式的时候的时候提示显存不足（内存不足？），我的电脑是r9000p 3060版本的，显存6G, 内存16G但是我又开了100G的虚拟内存。数据集大概有4Kw条，占20多G内存。怎么破呢，可以弄成迭代器吗，有无大佬给个demo，万分感谢。

我看的这个教程 https://m.imooc.com/wiki/tensorflow-tensorflowloadDateFrame

ApkB · 发表于 2021-11-10 22:09:24

这个超出MJJ理解范围了

期权小王子 · 发表于 2021-11-10 22:18:29

我有个例子

期权小王子 · 发表于 2021-11-10 22:19:52

你在读数据文件时，不要一次全部读到内存中

期权小王子 · 发表于 2021-11-10 22:20:58

做成迭代器就行

		自动登录	找回密码
密码			立即注册

求助tensorflow大佬

本帖子中包含更多资源