rsnt_translations
56,420
edits
MPI4py: Difference between revisions
Jump to navigation
Jump to search
no edit summary
No edit summary |
No edit summary |
||
| Line 206: | Line 206: | ||
<!--T:40--> | <!--T:40--> | ||
2. Write your submission script | 2. Write your submission script. | ||
{{File | {{File | ||
|name=submit-mpi4py-gpu.sh | |name=submit-mpi4py-gpu.sh | ||
| Line 235: | Line 235: | ||
<!--T:45--> | <!--T:45--> | ||
srun python use_cupy.py; | srun python use_cupy.py; | ||
}} | |||
<!--T:47--> | <!--T:47--> | ||
| Line 240: | Line 241: | ||
<!--T:48--> | <!--T:48--> | ||
Before submitting your job, it is important to test that your submission script will start without errors. You can do a quick test in an [[Running_jobs#Interactive_jobs|interactive job]]. | Before submitting your job, it is important to test that your submission script will start without errors. | ||
You can do a quick test in an [[Running_jobs#Interactive_jobs|interactive job]]. | |||
<!--T:49--> | <!--T:49--> | ||
4. Submit your job | 4. Submit your job | ||
{{Command|sbatch submit-mpi4py-gpu.sh}} | {{Command|sbatch submit-mpi4py-gpu.sh}} | ||
= Troubleshooting = <!--T:50-- | = Troubleshooting = <!--T:50-- | ||
== ModuleNotFoundError: No module named 'mpi4py' == | == ModuleNotFoundError: No module named 'mpi4py' == | ||
If <code>mpi4py</code> is not accessible, you may get the following error when importing it: | If <code>mpi4py</code> is not accessible, you may get the following error when importing it: | ||