GluonCV跑代码经常碰到的问题

在使用Gluoncv进行示例代码使用的发现,发现经常碰到一些问题,下面来总结一下:
1、某个module中属性不存在的问题.
常碰见bug说某个module中属性不存在的问题,这个问题的主要原因是mxnet和gluoncv版本的问题。解决的方案是使用pre版本进行安装:

pip install --pre mxnet-cu90   #or mxnet-cu80
pip install --pre -U gluoncv   #更新到gluoncv0.3

通过pre来安装的版本,是最新的1.3.0的mxnet和0.3的gluoncv
2、pooled_storage_manager.h:84: cudaMalloc failed: out of memory
跑示例代码的时候经常碰到cudaMalloc failed: out of memory,通过github上的回复,主要的解决方案为:
1、修改net.short, net.max_size的大小,这个可以解决问题
2、增加GPU的容量,也就是换GPU
3、减少batch_size
4、修改网络的隐藏层的大小(不太好改,一般直接使用其自带的网络结构)

其他问题再碰到后更新

你可能感兴趣的:(mxnet使用,python)