The following tools are available on GPU equipped computers and servers. They are configured as loadable modules that activate a Python 3.6 virtual environment and set the required library paths to load CUDA 9.0.

TensorFlow

module load tensorflow/1.12

PyTorch 1.0

module load pytorch/1.0