Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "04f9d2bf3d26e026244a13111d2f18bc95a5bb04"
Commit c9a60a98 authored by lishen's avatar lishen
Browse files

python调用时可以查询dcu_versino

parent 01ec9b4b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_binding/setup.py
View file @ c9a60a98
...@@ -328,7 +328,7 @@ def get_extensions(): ...@@ -328,7 +328,7 @@ def get_extensions():
def main(): def main():
setup( setup(
name='warpctc_pytorch', name='warpctc_pytorch',
version=get_version(os.environ.get('ROCM_HOME')), version=get_version(_find_rocm_home()),
description='Torch fuseop Computer Vision Foundation', description='Torch fuseop Computer Vision Foundation',
keywords='computer vision', keywords='computer vision',
packages=find_packages(), packages=find_packages(),
......
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