01 2021 档案

摘要:背景: pytorch从1.6版本开始,已经内置了torch.cuda.amp,采用自动混合精度训练就不需要加载第三方NVIDIA的apex库了。本文主要从三个方面来介绍AMP: 一.什么是AMP? 二.为什么要使用AMP? 三.如何使用AMP? 四. 注意事项 正文: 一.什么是AMP? 默认情况 阅读全文
posted @ 2021-01-22 19:51 jimchen1218 阅读(28848) 评论(1) 推荐(8)
摘要:搭建一套基于docker的开发训练环境 步骤如下: O、安装ubuntu18.04及NVidia驱动,CUDA,CUDNN 1.安装Ubuntu系统 制作ubuntu18.04系统盘 安装ubuntu过程,比较简单,可参考该链接:https://blog.csdn.net/baidu_3660242 阅读全文
posted @ 2021-01-15 10:40 jimchen1218 阅读(5263) 评论(1) 推荐(1)
摘要:ubuntu版本16.04 一、清华源 # 清华大学源 deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial main restricted universe multiverse deb http://mirrors.tuna.tsinghu 阅读全文
posted @ 2021-01-11 08:46 jimchen1218 阅读(8295) 评论(0) 推荐(0)
摘要:一、安装依赖包: sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler sudo apt-get install --n 阅读全文
posted @ 2021-01-10 12:53 jimchen1218 阅读(473) 评论(0) 推荐(0)
摘要:一、镜像操作 1)构建镜像 docker build -t REPOSITORY:TAG . 注意最后一个点[.]不能缺少,生成完:runoob/centos为REPOSITORY,6.7为TAG 2)获取镜像 sudo docker pull REPOSITORY[:TAG] 3)为镜像打个标签  阅读全文
posted @ 2021-01-08 15:13 jimchen1218 阅读(262) 评论(0) 推荐(0)
摘要:本文主要介绍使用docker+pycharm方式来搭建pytoch训练平台 操作流程如下: 一、宿主机docker环境安装 可参考该链接:NVidia-Docker2安装与常用命令 二、新建Docker镜像 有两种方式: 1、直接新建Dockerfile的方式生成 1)新建一个Dockerfile文 阅读全文
posted @ 2021-01-08 14:28 jimchen1218 阅读(3228) 评论(0) 推荐(0)
摘要:背景: DL训练框架采用Pytorch,推理框架使用Caffe,模型使用的是基于Facebook新出的MaskRCNN改进版,主要使用ADAS的视觉感知,包括OD,车道线,语义分割等网络。 整体框架: ​ 图1 Mask R-CNN整体架构 Mask R-CNN是一个非常灵活的框架,可以增加新的分支 阅读全文
posted @ 2021-01-06 23:08 jimchen1218 阅读(3168) 评论(0) 推荐(0)