OpenDroneMap-ODM/opendm/gpu.py

16 wiersze
342 B
Python
Czysty Zwykły widok Historia

2021-02-09 17:14:05 +00:00
from opendm import log
from repoze.lru import lru_cache
@lru_cache(maxsize=None)
def has_gpus():
import pyopencl
try:
platforms = pyopencl.get_platforms()
2021-02-10 19:00:43 +00:00
for p in platforms:
log.ODM_INFO("Found GPU device: %s" % p.name)
return len(platforms) > 0
2021-02-09 17:14:05 +00:00
except Exception as e:
return False