Tensorflow utilizing GPU or CPU computation is available on Koko. Partition information can be found here. Please use these instructions for allocating resources:
GPU
- Allocate resources with the command
salloc -p <partition> --gres=gpu- The gres:gpu number indicates how many GPUs you’d like to allocate, the default is 1.
- Find the node allocated using the command ‘squeue’ by your name, it will normally be close to the bottom for newly allocated resources.
- Connect to your allocated node with the command ‘srun –x11 –pty bash’.
- Load the modules required with the command ‘module load shared tensorflow-gpu/1.1.0’
- Start Python.
CPU
- Allocate resources with the command
salloc -p <partition>- Partition information can be found here.
- Find the node allocated using the command ‘squeue’ by your name, it will normally be close to the bottom for newly allocated resources.
- Connect to your allocated node with the command ‘srun –x11 –pty bash’.
- Load the modules required with the command ‘module load shared tensorflow/1.1.0’
- Start Python.