From 4c0cd6d3ee67dbd9e03c8aa25252a76079ed4ad7 Mon Sep 17 00:00:00 2001
From: "Edward Wang (EcoF)" <hack@fb.com>
Date: Tue, 14 Jun 2022 01:45:29 -0700
Subject: [PATCH] fix submodule imports by importing functions directly

Summary:
fixes two sporadic issues from missing attributes:

- breaking circular imports
- submodule not being imported explicitly

Reviewed By: ehhuang

Differential Revision: D37071652

fbshipit-source-id: 7566952e748cc7cca8de9cf31eeb9f25d3b047ae
---
 torchvision/extension.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/torchvision/extension.py b/torchvision/extension.py
index ea837c234d3..2b401714fd6 100644
--- a/torchvision/extension.py
+++ b/torchvision/extension.py
@@ -47,10 +47,10 @@ def _check_cuda_version():
     """
     if not _HAS_OPS:
         return -1
-    import torch
+    from torch.version import cuda
 
     _version = torch.ops.torchvision._cuda_version()
-    if _version != -1 and torch.version.cuda is not None:
+    if _version != -1 and cuda is not None:
         tv_version = str(_version)
         if int(tv_version) < 10000:
             tv_major = int(tv_version[0])
@@ -58,7 +58,7 @@ def _check_cuda_version():
         else:
             tv_major = int(tv_version[0:2])
             tv_minor = int(tv_version[3])
-        t_version = torch.version.cuda
+        t_version = cuda
         t_version = t_version.split(".")
         t_major = int(t_version[0])
         t_minor = int(t_version[1])