diff --git a/basicsr/data/degradations.py b/basicsr/data/degradations.py
index 14319605d..892826b26 100644
--- a/basicsr/data/degradations.py
+++ b/basicsr/data/degradations.py
@@ -5,7 +5,10 @@
 import torch
 from scipy import special
 from scipy.stats import multivariate_normal
-from torchvision.transforms.functional import rgb_to_grayscale
+try:
+    from torchvision.transforms.functional import rgb_to_grayscale
+except ImportError:
+    from torchvision.transforms.functional_tensor import rgb_to_grayscale
 
 # -------------------------------------------------------------------- #
 # --------------------------- blur kernels --------------------------- #
diff --git a/setup.py b/setup.py
index bc228e473..6b32dfaf3 100644
--- a/setup.py
+++ b/setup.py
@@ -74,9 +74,10 @@ def write_version_py():
 
 
 def get_version():
+    version_namespace = {}
     with open(version_file, 'r') as f:
-        exec(compile(f.read(), version_file, 'exec'))
-    return locals()['__version__']
+        exec(compile(f.read(), version_file, 'exec'), version_namespace)
+    return version_namespace['__version__']
 
 
 def make_cuda_ext(name, module, sources, sources_cuda=None):