xichen98cn commited on
Commit
9cd27c1
·
verified ·
1 Parent(s): e89b19f

Update frozenseg/modeling/pixel_decoder/ops/setup.py

Browse files
frozenseg/modeling/pixel_decoder/ops/setup.py CHANGED
@@ -47,11 +47,11 @@ def get_extensions():
47
  "-D__CUDA_NO_HALF_CONVERSIONS__",
48
  "-D__CUDA_NO_HALF2_OPERATORS__",
49
  ]
50
- else:
51
- if CUDA_HOME is None:
52
- raise NotImplementedError('CUDA_HOME is None. Please set environment variable CUDA_HOME.')
53
- else:
54
- raise NotImplementedError('No CUDA runtime is found. Please set FORCE_CUDA=1 or test it by running torch.cuda.is_available().')
55
 
56
  sources = [os.path.join(extensions_dir, s) for s in sources]
57
  include_dirs = [extensions_dir]
 
47
  "-D__CUDA_NO_HALF_CONVERSIONS__",
48
  "-D__CUDA_NO_HALF2_OPERATORS__",
49
  ]
50
+ # else:
51
+ # if CUDA_HOME is None:
52
+ # raise NotImplementedError('CUDA_HOME is None. Please set environment variable CUDA_HOME.')
53
+ # else:
54
+ # raise NotImplementedError('No CUDA runtime is found. Please set FORCE_CUDA=1 or test it by running torch.cuda.is_available().')
55
 
56
  sources = [os.path.join(extensions_dir, s) for s in sources]
57
  include_dirs = [extensions_dir]