There is a newer version of PyTorchPyTorch provides tensor computation with strong GPU acceleration and deep neural networks built on a tape-based autograd system.
Accessing PyTorch 0.4.1-foss-2018b-Python-3.6.6
To load the module for PyTorch 0.4.1-foss-2018b-Python-3.6.6 please use this command on the BEAR systems (BlueBEAR, BEARCloud VMs, and CaStLeS VMs):
module load PyTorch/0.4.1-foss-2018b-Python-3.6.6
There is a GPU enabled version of this module: PyTorch 0.4.1-fosscuda-2018b-Python-3.6.6
BEAR Apps Version
For more information visit the PyTorch website.
This version of PyTorch has a direct dependency on: foss/2018b mkl-dnn/0.16-foss-2018b Python/3.6.6-foss-2018b PyYAML/3.13-foss-2018b-Python-3.6.6
These versions of PyTorch are available on the BEAR systems (BlueBEAR, BEARCloud VMs, and CaStLeS VMs). These will be retained in accordance with our Applications Support and Retention Policy.
Last modified on 5th June 2019