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.
930 Commits
28 Branches
25 MB
4647 Download
Tree: 6667100638
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '6667100638'
${ noResults }
MegEngine/imperative/python/test/conftest.py

conftest.py 208 B
Raw Normal View History

fix(mge/pytest): remove __init__ in py-test GitOrigin-RevId: e16eae9cbcb4038bd1abbf4c6089d9cbb7cd4afc
4 years ago
​
​
fix(imperative): sync command queue at exit GitOrigin-RevId: c0d6aad27535b5a42d22124fc8c4662b02707a7c
4 years ago
​
​
​
fix(mge/pytest): remove __init__ in py-test GitOrigin-RevId: e16eae9cbcb4038bd1abbf4c6089d9cbb7cd4afc
4 years ago
fix(imperative): sync command queue at exit GitOrigin-RevId: c0d6aad27535b5a42d22124fc8c4662b02707a7c
4 years ago
​
​
​
 123456789101112 
  1. import os

  2. import sys

  3. 

  4. import pytest

  5. 

  6. from megengine.core._imperative_rt.imperative import sync

  7. 

  8. sys.path.append(os.path.join(os.path.dirname(__file__), "helpers"))

  9. 

  10. 

  11. def pytest_runtest_teardown():

  12.     sync()

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