预测模型训练占用的显存的小工具。人工智能 程序员 软件开发 gpt4