PyTorch
文章平均质量分 81
cg101202
这个作者很懒,什么都没留下…
展开
-
Anconda安装的pytorch依赖的cuda版本和系统cuda版本不一致问题
背景1.使用Anaconda源码配置环境2. 源码需要使用python setup.py来编译涉及cuda的拓展模块,如 nms,ROIPool,ROIAlign等等3. 系统的CUDA和Conda装的cudatoolkit版本不同问题符合上述背景条件或者类似条件,会导致一些奇怪的错误,例如:ImportError: ***/ATSS/atss_core/_C.cpython-36m-...原创 2019-12-23 14:03:06 · 10346 阅读 · 4 评论 -
配置历史版本Detectron遇到的问题
背景 最近想测试一个目标检测模型CBNet,该模型是基于Detectron改写的,但是又不是最新的Detectron,属于历史版本。 众所周知,Detectron基于caffe2,所以先得安装caffe2。官网有多种安装方式,常用 源码安装或者conda预构建包安装。遇到的问题 因为以前源码编译过,这次想偷个懒用conda直接安装caffe2(其实就是安装pytorch,后来整合进去...原创 2019-12-21 22:00:45 · 865 阅读 · 0 评论 -
Pytorch启程篇
说白了我就是蹭蹭热度!!!PyTorch Python中的张量和动态神经网络,具有强大的GPU加速功能。目录说白了我就是蹭蹭热度!!!PyTorch目录官方资料传送门五花八门的中文社区入门资料收集官方资料传送门官方手册中文版参考手册官方例程五花八门的中文社区PyTorch 中文社区 - ApacheCNPyTorch Tu...原创 2018-07-26 09:24:36 · 473 阅读 · 0 评论 -
Pytorch学习笔记(一) 使用PyTorch搭建神经网络的套路
自古套路得人心!!!PyTorch Python中的张量和动态神经网络,具有强大的GPU加速功能。目录自古套路得人心!!!PyTorch目录常见模块的导入构建网络模型自定义Net类并实例化使用顺序容器构建网络并实例化定义训练函数定义测试函数常见模块的导入import torchimport torch.nn as nn...原创 2018-07-29 20:19:59 · 18815 阅读 · 12 评论 -
Pytorch学习笔记(三)Pytorch的常用操作记录
查看Pytorch网络的各层输出(feature map)、权重、偏置在使用Pytorch的时候,我总是在想我怎么看每一层的输出、权重、偏置,下面记录我自己的方法,应该是方式多样,但是我现在的水平没办法评价好坏,如果有更好的办法欢迎留言告知,我会珍惜每次学习的机会!!!weight and bias# Method 1 查看Parameters的方式多样化,直接访问即可mode...原创 2018-08-04 14:41:03 · 5236 阅读 · 2 评论 -
Pytorch学习笔记(二)使用Pytorch的常见错误汇总
那些年我们一起踩过的坑 _, pred = output.data.cpu().topk(1, dim=1)RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorCopy.c:70...原创 2018-07-31 17:12:58 · 71857 阅读 · 7 评论 -
Pytorch学习笔记(四)各组件的Debug记录
Pytorch数据并行的流程定义:model = torch.nn.DataParallel(model, device_ids=args.gpus).cuda()在调用网络模型(类实例)时,实际是进入模型基类Module的特殊方法__call__,实现对类实例的调用(module.py)。再通过self.forward进入DataParallel类的forward方法(data_para...原创 2018-11-18 17:00:36 · 1525 阅读 · 0 评论 -
Jetson TX1安装 Pytorch
参考链接:https://gist.github.com/dusty-nv/ef2b372301c00c0a9d3203e42fd83426https://github.com/andrewadare/jetson-tx2-pytorchPyTorch for Jetson Nano - NVIDIA Developer Forumshttps://github.com/dusty-nv...原创 2019-06-06 13:10:32 · 1264 阅读 · 0 评论