杂项

控制线程数

内核映射 [1] 定义了输入特征矩阵的哪一行对应于输出特征矩阵的哪一行。 然而，随着维度的增加，这是一个昂贵的操作。 幸运的是，该操作的某些部分可以并行化，我们提供了一个多线程函数来加速此过程。

默认情况下，我们使用系统中所有可用的 CPU 线程。 但是，在某些情况下，这可能不是理想的。 只需定义一个环境变量 OMP_NUM_THREADS 来控制您想要使用的线程数。 例如， export OMP_NUM_THREADS=8; python your_program.py。 如果您使用 SLURM，环境变量 OMP_NUM_THREADS 将自动设置。

is_cuda_available

MinkowskiEngine.is_cuda_available() → bool

cuda_version

MinkowskiEngine.cuda_version() → int

get_gpu_memory_info

MinkowskiEngine.get_gpu_memory_info() → Tuple[int, int]

set_memory_manager_backend

MinkowskiEngine.set_memory_manager_backend(backend: MinkowskiEngineBackend._C.GPUMemoryAllocatorType)

set_gpu_allocator 的别名。 已弃用，将被删除。