MingzhenL's picture
Add files using upload-large-folder tool
01857e4 verified
Raw History Blame Contribute Delete
884 Bytes
#!/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