Difference between revisions of "PyTorch"

From UFRC
Jump to navigation Jump to search
Line 22: Line 22:
 
<!--Modules-->
 
<!--Modules-->
 
==Environment Modules==
 
==Environment Modules==
Run <code>module spider {{#var:app}}</code> to find out what environment modules are available for this application.
+
To use PyTorch on HiPerGator, you first need to load one of the PyTorch [[Modules_Basic_Usage|environment modules]].
 +
Run <code>module spider {{#var:app}}</code> to find out what environment modules are available for PyTorch.
 +
 
 +
Once you determine which module you want to use, load it as follows (using <code>pytorch/1.7.1</code> as an example):
 +
<code>
 +
module load pytorch/1.7.1
 +
</code>
  
 
==System Variables==
 
==System Variables==

Revision as of 02:17, 27 March 2021

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.