We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 44d925c commit fe2db3dCopy full SHA for fe2db3d
ml3d/metrics/__init__.py
@@ -1,6 +1,7 @@
1
import open3d
2
3
-if open3d._build_config['BUILD_CUDA_MODULE']:
+if open3d.core.cuda.device_count() > 0:
4
+ # Open3D is built with CUDA and the machine has a CUDA device.
5
from open3d.ml.contrib import iou_bev_cuda as iou_bev
6
from open3d.ml.contrib import iou_3d_cuda as iou_3d
7
else:
0 commit comments