submit_batch_jobs.sh not working in GPU environment

[mattgrayling]

I am working on deploying GPU jobs with submit_batch_jobs.sh in SNANA, but in the GPU environment submit_batch_jobs.sh is not working because of some import errors, for example ModuleNotFoundError: No module named 'coloredlogs'. I'm sure there will be more things down the line, but in the first instance could all the packages be added so that submit_batch_jobs.sh -h can run successfully?

[heather999]

Hi @mattgrayling
The GPU environment indeed has a smaller subset of packages than what we have in the CPU-only environment. I am not sure what list of packages submit_batch_jobs.sh needs explicitly (though I can imagine coloredlogs might be one of them) I can also ask Rick, but here are the full lists of what we install in the CPU environment:
the packages from conda-forge
and others from PyPI we pull in astrodash

Do you know which packages we really need? I'm hesitating to put everything into the GPU-enabled environment for fear that there will be some inevitable conflicts. Though we could try it and see what happens.