Merge branch 'master' of https://github.com/mit-han-lab/torchsparse

Author: kentang-mit

setup.py

@@ -1,11 +1,9 @@
 import torch
+import os
 from setuptools import find_packages, setup
 
-has_cuda = torch.cuda.is_available()
-if has_cuda:
- from torch.utils.cpp_extension import BuildExtension, CUDAExtension
-else:
- from torch.utils.cpp_extension import BuildExtension, CppExtension
+from torch.utils.cpp_extension import BuildExtension, CppExtension, CUDAExtension, CUDA_HOME
+has_cuda = (torch.cuda.is_available() and CUDA_HOME is not None) or os.getenv("FORCE_CUDA", "0") == "1"
 
 from torchsparse import __version__
 

torchsparse/utils/kernel_region.py

@@ -1,11 +1,9 @@
-import json
 import os
 import os.path
 import random
 import sys
 from collections import Sequence
 
-import h5py
 import numpy as np
 import torch