国内即百度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内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复