小编ray*_*ang的帖子

只有'导入keras'需要10GB的GPU

我正在使用带有预制AMI的EC2 GPU机器.我只是把'import keras'并运行它.需要11519MiB(12181MiB).这是无意义的吗?

我用谷歌搜索它.我猜它可能来自路径设置.我不知道哪个导致内存泄漏.

任何知道解决方案或经验与我相似的人?

--UPDATE--我公司的GPU服务器中的LD_LIBRARY_PATH当我第一次使用服务器时,只安装了tensorflow,而不是tensorflow-gpu.所以我做了'pip install --upgrade tensorflow-gpu' 在此输入图像描述

memory memory-leaks gpu keras tensorflow

2
推荐指数
1
解决办法
312
查看次数

标签 统计

gpu ×1

keras ×1

memory ×1

memory-leaks ×1

tensorflow ×1