hdk: 24.1.RC3
cann:8.0.RC3
torch:2.1.0
torch_npu:2.1.0.post8conda create -n PeptideBERT python=3.9cd PeptideBERT
pip install -r requirements.txtpip install torch==2.1.0
wget https://gitee.com/ascend/pytorch/releases/download/v6.0.rc3-pytorch2.1.0/torch_npu-2.1.0.post8-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
pip3 install torch_npu-2.1.0.post8-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl已存放置PeptideBERT/data目录(hemo、nf、sol)
source /usr/local/Ascend/ascend-toolkit/set_env.shpython train.py