你应该加载pytorch模型并将其作为全局变量。模型和任务的示例将在同一进程中运行

from celery.signals import worker_process_init
# for more information about worker_process_init, read here:
# https://docs.celeryproject.org/en/stable/userguide/signals.html#worker-process-init

pytorch_model = None

@worker_process_init.connect()
def init_worker_process(**kwargs):
    """
    load model before running tasks
    :param kwargs:
    :return:
    """
    global pytorch_model
    pytorch_model = load_model()

@app.task
def predict_task(image: np.ndarray):
    return pytorch_model.predict(image)

你可以开始celery 工人与“独奏”池类型
celery -A tasks worker --pool=solo --loglevel=info
Solo只创建一个线程并使用该线程运行celepie任务。此处不能提供并发数。