Machine learning / CUDA

Machine learning / CUDA2018-12-12T08:41:09+00:00

Machine learning libraries

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