python调用时可以查询dcu_versino

c9a60a98 · lishen · 01ec9b4b · c9a60a98
Commit c9a60a98 authored Jun 16, 2023 by lishen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pytorch_binding/setup.py pytorch_binding/setup.py +1 -1

No files found.
--- a/pytorch_binding/setup.py
+++ b/pytorch_binding/setup.py
@@ -328,7 +328,7 @@ def get_extensions():
 def main():
    setup(
        name='warpctc_pytorch',
-        version=get_version(os.environ.get('ROCM_HOME')),
+        version=get_version(_find_rocm_home()),
        description='Torch fuseop Computer Vision Foundation',
        keywords='computer vision',
        packages=find_packages(),