hdk：24.1.RC3
cann：8.2.RC1
python：3.10
torch：2.6.0
torch-npu：2.6.0

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

1. 新建conda环境
conda create -n pyehr-test python==3.11
conda activate pyehr-test

2. 安装依赖
pip install -r requirements.txt
pip install torch_npu==2.6.0 decorator attrs psutil absl-py cloudpickle ml-dtypes psutil scipy tornado pyyaml numpy==1.26.4 lightning==2.5.6 lightning-utilities==0.15.2

/root/miniconda3/envs/pyehr-test/lib/python3.11/site-packages/lightning/fabric/accelerators/cuda.py文件的166行加上return True

export CPU_AFFINITY_CONF=1
export TASK_QUEUE_ENABLE=2

cd datasets
wget https://static-content.springer.com/esm/art%3A10.1038%2Fs42256-020-0180-7/MediaObjects/42256_2020_180_MOESM3_ESM.zip
unzip 42256_2020_180_MOESM3_ESM.zip
cp time_series_* tjh/raw/
cd ..
python datasets/preprocess_tjh.py
# cdsl数据集需要在官网申请，这里只跑了tjh数据集
# 将dl.py中cdsl相关的删除即可，这里epoch只跑1轮

python train.py