Ejemplos de MinkowskiEngine.cuda_version en Python

Ejemplo n.º 1

Archivo: utility_functions.py Proyecto: BestSonny/MinkowskiEngineM

 def test(self):
     self.assertTrue(ME.is_cuda_available() == torch.cuda.is_available())
     if ME.is_cuda_available():
         print(ME.cuda_version())
         print(ME.get_gpu_memory_info())