训练时发现有一部分显存被占用

wanggan · 发表于 2021-1-4 02:41:34

星级打分

1
2
3
4
5

平均分:NAN 参与人数:0 我的评分:未评

2021-01-04 02:01:15.587321: E tensorflow/stream_executor/cuda/cuda_driver.cc:806] failed to allocate 2.13G (2287871488 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory
请问如何知道2.13G的显存被谁占用了？如何去释放这部分显存？我的电脑现在除了dfl别的都已经关掉了

hsTony · 发表于 2021-1-4 03:28:01

你这个是显存已经爆了，不是2.13G显存不能用，只是单纯的你的显卡不够罢了。你需要降低训练参数比如BS，在WIN10环境下，系统本身以及显存自身的缓冲区的原因，显存不可能百分百释放给你，你可考虑使用linux环境，但实际提升的也不多。

mzh920 · 发表于 2021-1-4 08:06:33

改成win7可以提升。以前看过这方面速度对比

solobabbit · 发表于 2021-1-4 12:42:13

降低数值是最好的解决办法。

gamecp9 · 发表于 2021-1-6 12:43:02

hsTony 发表于 2021-1-4 03:28
你这个是显存已经爆了，不是2.13G显存不能用，只是单纯的你的显卡不够罢了。你需要降低训练参数比如BS，在W ...

非常详细

anweishijian · 发表于 2021-1-15 19:27:47

提示: 作者被禁止或删除内容自动屏蔽

wanghao · 发表于 2021-3-5 15:58:38

闲逛路过，水丹

whiteli · 发表于 2021-3-28 18:48:33

真的不错

月亮是我啃弯的 · 发表于 2021-3-29 10:50:12

提示: 作者被禁止或删除内容自动屏蔽

想飞的翅膀 · 发表于 2021-3-29 15:25:53

路过支持一个

		自动登录	找回密码
密码			立即注册（仅限QQ邮箱）

训练时发现有一部分显存被占用

万事如意节日勋章

开心娱乐节日勋章