我是靠谱客的博主 优美电源,这篇文章主要介绍MegEngine 天元 深度学习框架发布,现在分享给大家,希望可以做个参考。

国内即百度paddlepaddle 发布之后,旷视科技又发布了MagEngine 深度学习框架。目前仅仅支持ubuntu 系统。python 3.5 以上。cuda 10.0。目前代码已在github上开源,地址是https://github.com/MegEngine

 

目前试验了一下物体检测的代码。在服务器上运行发现,多卡运行有点问题。

具体运行代码。

python3 tools/train.py -f retinanet_res50_1x_800size.py   -n 10 --batch_size 2 --dataset-dir

后续补充

最后

以上就是优美电源最近收集整理的关于MegEngine 天元 深度学习框架发布的全部内容,更多相关MegEngine内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(73)

评论列表共有 0 条评论

立即
投稿
返回
顶部