pytorch¶
Description¶
PyTorch is a GPU accelerated tensor computational framework with a Python front end. Functionality can be easily extended with common Python libraries such as NumPy, SciPy, and Cython. Automatic differentiation is done with a tape-based system at both a functional and neural network layer level. This functionality brings a high level of flexibility and speed as a deep learning framework and provides accelerated NumPy-like functionality.
NGC Catalog¶
https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
Available Versions on RCAC Clusters¶
| Cluster | Versions |
|---|---|
| ANVIL | 20.06-py3, 20.11-py3, 20.12-py3, 21.06-py3, 21.09-py3 |
| GILBRETH | 20.02-py3, 20.03-py3, 20.06-py3, 20.11-py3, 20.12-py3, 21.06-py3, 21.09-py3, 24.10-py3, 24.11-py3, 24.12-py3, 25.01-py3 |
| SCHOLAR | 20.02-py3, 20.03-py3, 20.06-py3, 20.11-py3, 20.12-py3, 21.06-py3, 21.09-py3 |
Module¶
You can load the container module by:
Note for using pytorch
Run module spider pytorch beforehand to check available versions and any prerequisite modules.