You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
Megvii Engine Team 5198b7834c fix(mge/functional): fix expand_dims for scalar 4 years ago
..
__init__.py chore(mge/functional): add compatible code for functional api 4 years ago
debug_param.py fix(mgb): fix spell error 4 years ago
distributed.py chore(mge): update copyright years 4 years ago
elemwise.py refactor(mge/functional): remove dependence to trace in functional implementations 4 years ago
external.py feat(imperative/src): python wrapper for cambricon and atlas runtime opr 4 years ago
inplace.py feat(mge/functional): add repeat and tile opr 4 years ago
loss.py feat(functional/loss): add reduction choices to loss functions 4 years ago
math.py feat(mgb/opr): add check_has_inf kernel and opr 3 years ago
metric.py feat(functional/ops): add _assert_equal 4 years ago
nn.py docs(syncbn): complete syncbn document 4 years ago
quantized.py refactor(mge): use lower case for default string parameters in functional and module 4 years ago
tensor.py fix(mge/functional): fix expand_dims for scalar 3 years ago
utils.py chore(mge/functional): add compatible code for functional api 4 years ago
vision.py fix(mgb/opr): correct nvof out shape computation 4 years ago

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境,不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序,请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉,欢迎访问 MegStudio 平台

C++ Cuda Python C SVG other