OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
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.
2710 Commits
28 Branches
25 MB
3464 Download
 
 
 
 
 
 
Tree: 32b31fd578
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '32b31fd578'
${ noResults }
MegEngine/imperative/python/megengine
Megvii Engine Team 32b31fd578 fix(mgb): change the check method of cuda sm code 2 years ago
..
amp fix(imperative/amp): add is_scalar check in elemwise and concat 3 years ago
autodiff fix(imperative/amp): fix format transformation for symbol trans 3 years ago
core fix(mge/tools): fix network_visualize for op without out shapes 2 years ago
data docs(dataloader): update dataloader docstring 2 years ago
distributed feat(mge/distributed): add support for batch send recv op 3 years ago
dtr chore(license): remove all license in file header 3 years ago
experimental chore(license): remove all license in file header 3 years ago
functional Revert "feat(dnn): add elemwise modes" 2 years ago
hub chore(license): remove all license in file header 3 years ago
jit feat(serialization): add no_change_graph and version param whem dump model 3 years ago
module feat(imperative/amp): add set_format 3 years ago
optimizer fix(imperative): fix inplace operation of optim 2 years ago
quantization chore(license): remove all license in file header 3 years ago
random chore(license): remove all license in file header 3 years ago
tools fix(mge/tools): fix network_visualize for op without out shapes 2 years ago
traced_module fix(traced_module): fix trace_module function may raise error in finally scope 3 years ago
utils fix(mge/tools): fix module_stats for duplicated module 2 years ago
__init__.py fix(mgb): change the check method of cuda sm code 2 years ago
_multistream.py perf(mge): improve cross stream memory borrowing 3 years ago
config.py fix(mge): control compute_mode by context 3 years ago
conftest.py ci(pytest): fixture doctest namespace in pytest config 3 years ago
device.py feat(mge/device): enable to get cuda/cudnn/tensorrt version 2 years ago
logger.py chore(license): remove all license in file header 3 years ago
serialization.py chore(license): remove all license in file header 3 years ago
tensor.py feat(imperative/amp): add set_format 3 years ago
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

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

C++ Cuda Python C SVG other