This shows you the differences between two versions of the page.
| Both sides previous revision Previous revision Next revision | Previous revision |
| software:tensorflow:caviness [2019-09-03 17:08] – anita | software:tensorflow:caviness [2021-04-27 16:21] (current) – external edit 127.0.0.1 |
|---|
| </code> | </code> |
| |
| For example, say your TensorFlow Python script is called ''tf-script.py'', then you should copy this file or create it the ''tf-test-001'' directory, then copy the tensorflow.qs job script template: | For example, say your TensorFlow Python script is called ''tf-script.py'', then you should copy this file or create it in the ''tf-test-001'' directory, then copy the tensorflow.qs job script template: |
| |
| <code bash> | <code bash> |
| $ cp /opt/templates/slurm/applications/tensorflow.qs . | $ cp /opt/shared/templates/slurm/applications/tensorflow.qs . |
| </code> | </code> |
| |
| You will need to modify the copy of ''tensorflow.qs'' accordingly (''%%-%%-cpus-per-task=2'' to however many CPU cores you need, 1 - 36; ''%%-%%-mem-per-cpu=1024M'' to alter max memory limit; ''%%-%%-job-name=tensorflow'', etc.). The template has extensive documentation that should assist you in customizing it for the job. Last but not least, the last line should be changed to match your Python script name and for this example, it it would be ''tf-script.py'': | You will need to modify the copy of ''tensorflow.qs'' accordingly such as ''%%-%%-cpus-per-task=2'' to however many CPU cores you need, 1 - 36; ''%%-%%-mem-per-cpu=1024M'' to alter max memory limit; ''%%-%%-job-name=tensorflow'', etc. The template has extensive documentation that should assist you in customizing it for the job. Last but not least, you need to specify the version of Tensorflow you want via VALET, and then the last line should be changed to match your Python script name and for this example, so for this example it would be ''tf-script.py'': |
| |
| |
| <code bash> | <code bash> |
| | ... |
| | |
| | # |
| | # Add a TensorFlow container to the environment: |
| | # |
| | vpkg_require tensorflow/1.12.0:py3 |
| | |
| # | # |
| # Execute our TensorFlow Python script: | # Execute our TensorFlow Python script: |
| # | # |
| python tf-script.py | python3 tf-script.py |
| </code> | </code> |
| |