PyTorch

From UFRC
Revision as of 02:21, 27 March 2021 by Stuckyb (talk | contribs)
Jump to navigation Jump to search

Description

pytorch website  

PyTorch is a machine learning library with strong support for neural networks and deep learning. PyTorch also has a large user base and software ecosystem.

Environment Modules

To use PyTorch on HiPerGator, you first need to load one of the PyTorch environment modules. Run module spider pytorch to find out what environment modules are available for PyTorch.

Once you determine which module you want to use, load it as follows (using pytorch/1.7.1 as an example):

module load pytorch/1.7.1

System Variables

  • HPC_PYTORCH_DIR - installation directory
  • HPC_PYTORCH_BIN - executable directory



Job Script Examples

See the PyTorch_Job_Scripts page for pytorch Job script examples.