Commit 0fa8f7af authored by xiabo's avatar xiabo
Browse files

适配rocm

parent 86d9b043
......@@ -96,7 +96,7 @@ def get_version():
version_file = 'mmcv/version.py'
with open(version_file, encoding='utf-8') as f:
exec(compile(f.read(), version_file, 'exec'))
return locals()['__version__']
return locals()['__dcu_version__']
def parse_requirements(fname='requirements/runtime.txt', with_version=True):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment