hdk:24.1.RC3
cann:8.2.RC1
python:3.10
torch:2.6.0
torch-npu:2.6.0git clone https://atomgit.com/panxiaomu/esm3.git1. 新建conda环境
conda create --name esm3 python=3.11
conda activate esm3
2. 安装依赖
cd esm
pip install esm
pip install cloudpickle ml-dtypes psutil tornado absl-py httpx numpy==1.26.3
pip install torch_npu==2.6.0下载huggingface上的全部内容,放到/root/.cache/torch/hub/checkpoints/esm3-sm-open-v1目录下
https://huggingface.co/EvolutionaryScale/esm3-sm-open-v1/tree/mainexport CPU_AFFINITY_CONF=1
export TASK_QUEUE_ENABLE=2python test-esm3.py