From 4c0cd6d3ee67dbd9e03c8aa25252a76079ed4ad7 Mon Sep 17 00:00:00 2001 From: "Edward Wang (EcoF)" 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])