pytorch
hajungong007
这个作者很懒,什么都没留下…
展开
-
windows上部署libtorch
ref:https://zhuanlan.zhihu.com/p/68901339https://zhuanlan.zhihu.com/p/55292836原创 2019-11-02 23:52:05 · 785 阅读 · 0 评论 -
PyTorch转TensorRT
安装TensorRT按照官网的安装教程即可,我装的是TensorRT-6.0.1.5。安装onnxsudo apt-get install protobuf-compiler libprotoc-devpip install onnx==1.5.0安装 onnx-tensorrtgit clone --recursive https://github.com/onnx/onn...原创 2019-10-27 17:51:13 · 2733 阅读 · 0 评论 -
pytorch在同一台服务器上同时训练多个任务
使用pytoch在同一台服务器上同时训练多个任务时会出现以下问题:File "./torch-dist/mnist-dist.py", line 201, in <module> init_processes(args.rank, args.world_size, run, args.batch_size, backend=args.backend) File "./to...原创 2019-09-24 14:37:07 · 8467 阅读 · 2 评论 -
libtorch推理实现
已经完成了pytorch C++的分类、语义分割、目标检测等任务的推理实现,在使用过程中遇到opencv4.0以上版本与libtorch不兼容的问题,是官方libtorch默认使用旧版c++11 ABI编译导致的,需要重新源码编译libtorch。参考了https://oldpan.me/archives/pytorch-c-libtorch-inference,以后有时间再完善本博客。...原创 2019-08-27 10:24:15 · 1976 阅读 · 3 评论 -
pytorch教程
https://github.com/stared/thinking-in-tensors-writing-in-pytorch原创 2019-07-22 00:03:38 · 111 阅读 · 0 评论 -
PyTorch深度网络逐层性能分析器
ref: https://github.com/awwong1/torchprof原创 2019-07-20 23:03:43 · 2310 阅读 · 0 评论 -
Pytorch网络结构可视化
安装可以通过以下的命令进行安装pip install graphvizpip install torch torchvisionpip install tensorwatch载入库import sysimport torchimport tensorwatch as twimport torchvision.models网络结构可视化model = torchvision...原创 2019-06-15 20:39:16 · 6619 阅读 · 10 评论 -
THOP: 估算PyTorch模型的MACs 模块
THOP: PyTorch-OpCounterHow to installThrough PyPipip install thopUsing GitHub (always latest)pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.gitHow to useBasic us...原创 2019-05-13 11:24:10 · 15466 阅读 · 8 评论 -
pytorch计算模型的显存占用率和节省内存技巧
计算模型的显存占用率如何计算模型以及中间变量的显存占用大小可以参考此文。如何在Pytorch中精细化利用显存,牺牲计算速度减少显存用量,将计算过程分为两半,先计算一半模型的结果,保存中间结果再计算后面一半的模型。再次浅谈Pytorch中的显存利用问题(附完善显存跟踪代码),借用Pytorch-Memory-Utils这个工具来检测我们在训练过程中关于显存的变化情况,分析出我们如何正确释放多...原创 2019-01-19 23:15:12 · 11059 阅读 · 0 评论 -
python、pytorch调试工具——ipdb
1、安装:pip install ipdb2、使用:在代码开头import ipdb,如下:try: import ipdbexcept: import pdb as ipdb在需要开始调试的地方加入语句:ipdb.set_trace()当代码执行到上面那条语句时,会进入debug模式。3、常用命令:n (下一个),next单步执行EN...原创 2018-05-19 23:47:57 · 5915 阅读 · 0 评论 -
How to use Tensorboard with PyTorch
Let's directly dive in. The thing here is to use Tensorboard to plot your PyTorch trainings. For this, I use TensorboardX which is a nice interface communicating Tensorboard avoiding Tensorflow depend...转载 2018-04-08 17:32:43 · 383 阅读 · 0 评论