hdk: 25.0.rc1.1
cann：8.1.RC1
torch：2.3.1
torch_npu：2.3.1.post2

conda create -n esm_2 python=3.11.6

pip install pyyaml numpy decorator attrs matplotlib

pip install torch==2.3.1

wget https://gitee.com/ascend/pytorch/releases/download/v7.0.0-pytorch2.3.1/torch_npu-2.3.1.post2-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

pip3 install torch_npu-2.3.1.post2-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

pip3 install fair-esm
pip install "fair-esm[esmfold]"

source /usr/local/Ascend/ascend-toolkit/set_env.sh

python pretrained.py