请问，如何优化pytorch的模型预测速度

用deeplab v3 + 模型训练好之后，加载模型训练，batchsize设为1，预测单张图片（1024*768）速度很慢，要大概7-8s有什么方法可以加快预测速度吗？