kopia lustrzana https://github.com/OpenDroneMap/ODM
20 wiersze
450 B
Python
20 wiersze
450 B
Python
from opendm import log
|
|
from repoze.lru import lru_cache
|
|
|
|
@lru_cache(maxsize=None)
|
|
def has_gpus():
|
|
try:
|
|
import pyopencl
|
|
except:
|
|
log.ODM_INFO("PyOpenCL is missing (not a GPU build)")
|
|
return False
|
|
|
|
try:
|
|
platforms = pyopencl.get_platforms()
|
|
for p in platforms:
|
|
log.ODM_INFO("Found GPU device: %s" % p.name)
|
|
|
|
return len(platforms) > 0
|
|
except Exception as e:
|
|
return False
|