GPU 环境配置

用于GPU加速模型训练