TensorFlow on the Grid

From CLSP Wiki
Revision as of 11:45, 17 September 2018 by Paxia (Talk | contribs)

(diff) ←Older revision | view current revision (diff) | Newer revision→ (diff)
Jump to: navigation, search

Setup

The CPU version of TensorFlow does not need any special setup -- the commands for installing on Ubuntu should also work on the grid.

For GPU, our current version of CUDA on the grid is 9.1, while the compiled versions expect CUDA9.0. To get around this, follow the instructions to install the built versions at https://github.com/mind/wheels/releases.

Tips

If your program crashes with a warning stating that the GPU is already in use, this may be because you have a lot of CPU-based computation before using the GPU (assuming you reserved a GPU according to GPUs on the Grid). To get around this, you can claim a GPU earlier by making a tf.Session object as soon as possible in your code.