首页 / 浏览问题 / 桌面GIS / 问题详情
SuperMap iDesktopX 10i 机器学习模型训练好像没有用GPU?
2EXP 2020年08月12日

使用产品:SuperMap iDesktopX 10i  操作系统:winserver 2019
数据类型: 超图深度学习实例数据(resources_ml中的plane.tif、label.udbx)

问题描述:1、本机带有显卡,自己基于anaconda 使用requirements-conda-gpu.yml配置文件已新建了相关的gpu环境

                                  2、 使用 示例数据plane.tif、label.udbx做训练数据生成,生成训练数据后使用模型训练;模型训练可以正常使用,但是CUDA使用率很低,一直只有百分之10几,感觉没有用GPU?

               

    3、同步验证tensorflow-gpu是否没有装好,使用iobjectspy-gpu环境用自己写的语义分割网络做训练验证,CUDA使用率很高

    

SuperMap iDesktopX 中的目标监测模型训练为什么CUDA使用率不高???

1个回答

idesktopx切换python环境后,需要重启下,然后在python控制台查看下当前tf环境是否是gpu。
1,505EXP 2020年08月12日

是 gpu的,语义分割可以GPU可以利用起来,目标监测的GPU跑不起来,想问下目标监测GPU跑不起来的原因?是不是目标监测的batchsize只能设置1导致的?好像不能让设置大?我显卡很好啊双2080Ti

...