已开启
deepchem支持pytorch 2.5.1+ 版本 #1
iambowen1984创建于  2月4日
iambowen1984成员
2月4日 创建

🚀 Feature

升级deepchem 框架依赖的torch版本到2.5.1+,减少迁移到torch npu的代码量。

Motivation

DeepChem框架适配

Additional context

验证条件:

  1. 基于setup.py可以构建出deepchem pip包;
  2. 安装发布的deepchem包之后,可以运行基础的torch model,选择分类模型和一个GNN模型,可基于CPU/GPU进行验证。
likedislike
Iiambowen1984成员
2月4日 将 TokenDance 设为负责人
Iiambowen1984成员
2月4日 添加了label:enhancement
iambowen1984成员
2月4日 评论:

当前DGL不支持torch > 2.5的版本,先暂停升级,先基于旧版本进行验证,如出现因版本导致的问题,升级到DGL和deepchem都兼容的torch版本 < 2.5

likedislike