File size: 884 Bytes
01857e4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | #!/bin/bash
#SBATCH -A p33139 ## 你的账户
#SBATCH -p gengpu ## GPU 分区
#SBATCH --gres=gpu:a100:1 ## 请求一块 A100 GPU
#SBATCH -N 1 ## 请求一个节点
#SBATCH -n 1 ## 请求一个任务
#SBATCH -t 48:00:00 ## 最大运行时间 48 小时
#SBATCH --mem=80G
#SBATCH -o logs/msd_all_tasks_%j.out
#SBATCH -e logs/msd_all_tasks_%j.err
# 加载 Conda 环境
source /projects/p32685/miniconda3/etc/profile.d/conda.sh
conda activate msd_preprocess # 激活 Conda 环境（Python 3.10+, 含所有依赖）
# 切换到脚本目录
cd /projects/p32685/VESSA_data_preprocess
# 创建日志目录
mkdir -p logs
# 运行批量处理脚本，处理所有 10 个 MSD Task
# 输入数据位置: /projects/p32685/MSD/TaskXX_*
# 输出位置: /projects/p32685/RL_dataset/MSD_taskXX/
python -u process_all_msd_tasks.py --device cuda:0
|