update dcu_version

0811db01 · panning · afdca1a1 · 0811db01
Commit 0811db01 authored Jun 07, 2023 by panning
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

torchvision/__init__.py torchvision/__init__.py +4 -1

No files found.
--- a/torchvision/__init__.py
+++ b/torchvision/__init__.py
@@ -16,7 +16,10 @@ try:
    from .version import __version__  # noqa: F401
 except ImportError:
    pass
-
+try:
+    from .version import __dcu_version__  # noqa: F401
+except ImportError:
+    pass
 # Check if torchvision is being imported within the root folder
 if (not _HAS_OPS and os.path.dirname(os.path.realpath(__file__)) ==
        os.path.join(os.path.realpath(os.getcwd()), 'torchvision')):