Tensorflow

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.
  • SSH to your allocated node with the command ‘ssh <node> -X’.
  • 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.
  • SSH to your allocated node with the command ‘ssh <node> -X’.
  • Load the modules required with the command ‘module load shared tensorflow/1.1.0’
  • Start Python.
Posted in How-To-Guides, HPC