Resource usage prediction for deep learning model

Abstract

According to implementations of the subject matter described herein, there is provided a solution for predicting the resource usage of the deep learning model. In this solution, information about a deep learning model is obtained, the information comprising first information for describing the deep learning model and second information about an operating environment of a job associated with the deep learning model. The static resource usage of the job is determined based on the first information and a strategy of the job during runtime in the operating environment is determined. Afterwards, resource usage of the job during runtime in the operating environment is predicted based on the strategy and the static resource usage. With this solution, the usage of various resources of the deep learning model, such as computation power consumption, memory consumption, execution time, and the like, under a specific runtime strategy can be accurately predicted.