博客
关于我
工程实践_LFFD模型训练过程疑难杂症debug
阅读量:537 次
发布时间:2019-03-08

本文共 2359 字,大约阅读时间需要 7 分钟。

1.MxNet版本的LFFD需要安装CUDA10.1版本和CuDNN

若不满足会出现如下问题:

安装的CUDA版本太低或没有安装:

raceback (most recent call last):  File "configuration_10_320_20L_5scales_v2.py", line 17, in 
import mxnet File "/usr/local/lib/python3.6/dist-packages/mxnet/__init__.py", line 24, in
from .context import Context, current_context, cpu, gpu, cpu_pinned File "/usr/local/lib/python3.6/dist-packages/mxnet/context.py", line 24, in
from .base import classproperty, with_metaclass, _MXClassPropertyMetaClass File "/usr/local/lib/python3.6/dist-packages/mxnet/base.py", line 213, in
_LIB = _load_lib() File "/usr/local/lib/python3.6/dist-packages/mxnet/base.py", line 204, in _load_lib lib = ctypes.CDLL(lib_path[0], ctypes.RTLD_LOCAL) File "/usr/lib/python3.6/ctypes/__init__.py", line 348, in __init__ self._handle = _dlopen(self._name, mode)OSError: libcudart.so.10.1: cannot open shared object file: No such file or directory

没有安装CuDNN:

terminate called after throwing an instance of 'dmlc::Error'  what():  [20:48:36] ../include/mshadow/./stream_gpu-inl.h:173: Check failed: err == CUDNN_STATUS_SUCCESS (4 vs. 0) : CUDNN_STATUS_INTERNAL_ERRORAborted (core dumped)

2.正确使用Python和正确安装MxNet版本

若已经正确安装CUDA和CUDNN,仍然出现:

terminate called after throwing an instance of 'dmlc::Error'  what():  [20:48:36] ../include/mshadow/./stream_gpu-inl.h:173: Check failed: err == CUDNN_STATUS_SUCCESS (4 vs. 0) : CUDNN_STATUS_INTERNAL_ERRORAborted (core dumped)

有两种可能:首先查看MxNet版本是否正确,再在configuration_10_560_25L_8scales_v1.py代码中将如下代码注释:

# add mxnet python path to path env if needmxnet_python_path = '/home/heyonghao/libs/incubator-mxnet/python'sys.path.append(mxnet_python_path)

我们只需要使用我们本地默认的Python就行。

3.正确安装OpenCV

如出现如下问题:

During handling of the above exception, another exception occurred:Traceback (most recent call last):  File "_ctypes/callbacks.c", line 234, in 'calling callback function'  File "/root/work/mxnet/python/mxnet/operator.py", line 1052, in backward_entry    print('Error in CustomOp.backward: %s' % traceback.format_exc())UnicodeEncodeError: 'ascii' codec can't encode characters in position 369-376: ordinal not in range(128)

说明OpenCV版本没有正确安装,删除旧版本之后安装如下版本:

pip install opencv-python==3.4.5.20

4.正确设置batch_size

遇到如下问题,很可能是batch_size设置的太大:

MXNetError: cudaMalloc retry failed: out of memory

可以设置batch_size=16

转载地址:http://gsbiz.baihongyu.com/

你可能感兴趣的文章
MySQL 中文问题
查看>>
MySQL 中日志的面试题总结
查看>>
mysql 中的all,5分钟了解MySQL5.7中union all用法的黑科技
查看>>
MySQL 中的外键检查设置:SET FOREIGN_KEY_CHECKS = 1
查看>>
Mysql 中的日期时间字符串查询
查看>>
mysql 中索引的问题
查看>>
MySQL 中锁的面试题总结
查看>>
MySQL 中随机抽样:order by rand limit 的替代方案
查看>>
MySQL 为什么需要两阶段提交?
查看>>
mysql 为某个字段的值加前缀、去掉前缀
查看>>
mysql 主从
查看>>
mysql 主从 lock_mysql 主从同步权限mysql 行锁的实现
查看>>
mysql 主从互备份_mysql互为主从实战设置详解及自动化备份(Centos7.2)
查看>>
mysql 主从关系切换
查看>>
MYSQL 主从同步文档的大坑
查看>>
mysql 主键重复则覆盖_数据库主键不能重复
查看>>
Mysql 事务知识点与优化建议
查看>>
Mysql 优化 or
查看>>
mysql 优化器 key_mysql – 选择*和查询优化器
查看>>
MySQL 优化:Explain 执行计划详解
查看>>