一种基于PyTorch的高效柔性分子对接方法及装置

本发明公开一种基于PyTorch的高效柔性分子对接方法及装置，属于人工智能药物发现技术领域；对接方法包括：加载配置参数，并设置计算设备；解析配体文件和蛋白质文件，获得配体和蛋白质数据；创建Vina评分模型、变换模块，并初始化搜索算法；基于初始化搜索算法，结合配体和蛋白质数据，生成初始构象参数，并基于Vina评分模型、变换模块进行梯度下降优化，生成多个优化后的配体构象坐标、蛋白质柔性残基坐标以及对应的能量评分，构建候选构象集合；计算候选构象集合中构象间的RMSD，过滤冗余构象，选择最优构象集合，并保存最优构象的PDBQT文件，记录对应的能量评分信息，生成对接报告。利用PyTorch深度学习框架和GPU的并行计算能力，加速分子对接过程。