From ff21248496947b7f5df8b058e8a20b2b75173688 Mon Sep 17 00:00:00 2001 From: Mamat Shamshiev Date: Thu, 3 Feb 2022 12:23:23 +0300 Subject: [PATCH] remove setup runtime cuda calculations --- setup.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/setup.py b/setup.py index caf7007..46c15ec 100644 --- 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__)