【解决方案】tensorflow显存占比高可是GPU利用率低的问题

背景 在跑代码的时候,运行cmd,输入nvidia-smi,发现这样一个问题,显存的占比高特别低可是GPU利用率特别低。 web 解决方法 这个跟本身写的代码有关系,个人主要问题在于,读取磁盘中数据的时候,是依靠CPU在读取数据(此时磁盘读写会很是高,打开任务管理器就知道),同时,tensorflow在运行模型训练的时候,会从内存中读取数据,GPU训练会比较快,而CPU一直在从磁盘中读数据到内存中
相关文章
相关标签/搜索