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

浙公网安备 33010602011771号