Skip to content
Snippets Groups Projects
Select Git revision
  • fa99171e47a3e1690c2ed5071857d4edd9c23ef0
  • main default protected
2 results

run.sbatch

Blame
  • DataExtraction.sh 1.42 KiB
    #!/bin/bash -x
    ## Controlling Batch-job
    #SBATCH --account=deepacf
    #SBATCH --nodes=1
    #SBATCH --ntasks=13
    ##SBATCH --ntasks-per-node=13
    #SBATCH --cpus-per-task=1
    #SBATCH --output=DataExtraction-out.%j
    #SBATCH --error=DataExtraction-err.%j
    #SBATCH --time=05:00:00
    #SBATCH --partition=devel
    #SBATCH --mail-type=ALL
    #SBATCH --mail-user=b.gong@fz-juelich.de
    
    
    jutil env activate -p deepacf
    
    # Name of virtual environment 
    VIRT_ENV_NAME="virt_env_hdfml"
    
    # Loading mouldes
    source ../env_setup/modules_preprocess.sh
    # Activate virtual environment if needed (and possible)
    if [ -z ${VIRTUAL_ENV} ]; then
       if [[ -f ../${VIRT_ENV_NAME}/bin/activate ]]; then
          echo "Activating virtual environment..."
          source ../${VIRT_ENV_NAME}/bin/activate
       else 
          echo "ERROR: Requested virtual environment ${VIRT_ENV_NAME} not found..."
          exit 1
       fi
    fi
    
    # Declare path-variables
    source_dir="/p/fastdata/slmet/slmet111/met_data/ecmwf/era5/nc/"
    dest_dir="/p/scratch/deepacf/video_prediction_shared_folder/extractedData/"
    
    year="2010"
    
    # Run data extraction
    srun python ../../workflow_parallel_frame_prediction/DataExtraction/mpi_stager_v2.py --source_dir ${source_dir}/${year}/ --destination_dir ${dest_dir}/${year}/
    
    
    
    # 2tier pystager 
    #srun python ../../workflow_parallel_frame_prediction/DataExtraction/main_single_master.py --source_dir /p/fastdata/slmet/slmet111/met_data/ecmwf/era5/nc/${year}/ --destination_dir ${SAVE_DIR}/extractedData/${year}