Pytorch
高性能计算研发苦行僧
希望和各位大佬交流
展开
-
Pytorch 训练一些要点记录
我们用Pytroch 训练神经网络模型时,如果数据量稍大,或者网络过深,会造成训练速度过慢,这时,我们就需要保存中间结果,然后下次的时候,从中间结果恢复参数进行继续训练。记录一下代码的主要写法:import torch#导入 torch模块import argparse# 一个命令行参数处理很好用的Python 库parse = argparse.ArgumentParser(descriptio...原创 2018-07-16 14:17:12 · 3787 阅读 · 1 评论 -
Pytorch faster rcnn 三个系统版本运行情况说明
可能由于闲来无聊,我们将windows linux macos 三大系统的faster Rcnn 的pytorch 版本配置好了。然后做了性能比较。由于MacOS 采用N卡的web driver 所以确实可以看到,性能很渣,实时检测来说,只能达到 8-9fps. win 下可以达到 14-15 而Ubuntu 下就可以达到17-18fps 比win 10 下略高。可能由于ubuntu...原创 2018-09-07 15:59:54 · 1388 阅读 · 6 评论 -
MacOS 10.14.3 faster-rcnn 编译
由于MacOS 10.14.3 N卡驱动始终得不到解决。所以现记录CPU版本的编译过程首先需要编译PytorchGitHub上有一个编译好的Pytorch版本 pytorch-1.0 cuda 9.1的版本,我们可以直接进行安装。随后我们编译https://github.com/jwyang/faster-rcnn.pytorch这个版本的faster rcnn需要注意的地方是...原创 2019-01-29 23:36:49 · 302 阅读 · 0 评论 -
win 10 mmdetection 配置
-----2019-7-22 更新mmdetection 维护人员看来很用心啊,已经适配到了最新的pytorch 1.1 并且修改了编译的方式,看来是要官方支持Windows了,但是目前win下有两个bug 。sigmoid_focal_loss 和 mask 那个层的编译有问题。官方作者在github 上已经给予了回复。貌似是torch的bug目前如果想用的话,只能先注释掉相关的代码...原创 2019-04-11 11:01:26 · 4404 阅读 · 6 评论 -
pytorch 查看cuda 版本
由于pytorch的whl 安装包名字都一样,所以我们很难区分到底是基于cuda 的哪个版本。有一条指令可以查看import torchprint(torch.version.cuda)原创 2019-07-08 16:13:12 · 25760 阅读 · 1 评论