remove setup runtime cuda calculations

diagnocat · Feb 3, 2022 · ff21248 · ff21248
1 parent fb3872d
commit ff21248
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -8,8 +8,6 @@
 from torch.utils.cpp_extension import BuildExtension, CppExtension, CUDAExtension
 from setuptools import setup, find_packages
 
-if torch.cuda.is_available():
-    assert torch.matmul(torch.ones(2097153,2).cuda(),torch.ones(2,2).cuda()).min().item()==2, 'Please upgrade from CUDA 9.0 to CUDA 10.0+'
 
 this_dir = os.path.dirname(os.path.realpath(__file__))
 torch_dir = os.path.dirname(torch.__file__)