Spaces:
Runtime error
Runtime error
File size: 643 Bytes
d17e641 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import sys
import os
import subprocess
sim_pack = sys.argv[1]
task = sys.argv[2]
com_cap = sys.argv[3]
prec = sys.argv[4]
N_arr = range(6, 36, 2)
for N in N_arr:
with open('{}_{}_{}_{}.sh'.format(task, com_cap, prec, N), 'a') as job_file:
job_file.write('#!/bin/bash -l\n')
job_file.write('. /app/etc/profile.d/conda.sh\n')
if com_cap == 'st' or com_cap == 'mt':
job_file.write('conda activate sim_{}\n'.format(sim_pack))
elif com_cap == 'gpu':
job_file.write('conda activate sim_{}_gpu\n'.format(sim_pack))
job_file.write('python3 /home/{}_n{}.py\n'.format(task, N))
|