AI4S/chgnet
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

组件版本

hdk:25.2.0
cann:8.3.RC1
python:3.11
torch:2.7.1
torch-npu:2.7.1

拉取模型代码

git clone https://atomgit.com/AI4Science/chgnet.git

环境准备

1. 新建conda环境
conda create --name chgnet python=3.11
conda activate chgnet

2. 安装依赖
pip install chgnet
pip install pyyaml torch_npu==2.7.1 decorator attrs psutil absl-py cloudpickle ml-dtypes scipy tornado

推理

python npu_test.py