Fix celery tasks

We need to import the GPU class once so that sqlalchemy can properly set
up its internal data structures