nikigoli commited on
Commit
187246f
1 Parent(s): a39ebe0

Removed CUDA_HOME check

Browse files
Files changed (1) hide show
  1. models/GroundingDINO/ops/setup.py +2 -2
models/GroundingDINO/ops/setup.py CHANGED
@@ -41,7 +41,7 @@ def get_extensions():
41
  print("inside get_extensions")
42
  print(torch.cuda.is_available())
43
  print(CUDA_HOME)
44
- if torch.cuda.is_available() and CUDA_HOME is not None:
45
  extension = CUDAExtension
46
  sources += source_cuda
47
  define_macros += [("WITH_CUDA", None)]
@@ -52,7 +52,7 @@ def get_extensions():
52
  "-D__CUDA_NO_HALF2_OPERATORS__",
53
  ]
54
  else:
55
- raise NotImplementedError('Cuda is not availabel')
56
 
57
  sources = [os.path.join(extensions_dir, s) for s in sources]
58
  include_dirs = [extensions_dir]
 
41
  print("inside get_extensions")
42
  print(torch.cuda.is_available())
43
  print(CUDA_HOME)
44
+ if torch.cuda.is_available():
45
  extension = CUDAExtension
46
  sources += source_cuda
47
  define_macros += [("WITH_CUDA", None)]
 
52
  "-D__CUDA_NO_HALF2_OPERATORS__",
53
  ]
54
  else:
55
+ raise NotImplementedError('Cuda is not available')
56
 
57
  sources = [os.path.join(extensions_dir, s) for s in sources]
58
  include_dirs = [extensions_dir]