When the pumping unit is working, the actual load of each motor is different. How to adjust the actual input voltage of the motor according to the actual load of the pumping unit during the working process can significantly improve the efficiency of the pumping unit. This paper uses the neural network prediction algorithm to predict the actual input voltage of the motor, discusses the influence of the parameters of the neural network prediction algorithm on the prediction results, and compares the predicted value obtained by the neural network prediction algorithm with the measured value. The results show that the neural network prediction algorithm proposed in this paper can predict the optimal working voltage of the motor.