Hprc banner tamu.png

Difference between revisions of "SW:PyTorch"

From TAMU HPRC
Jump to: navigation, search
(Created page with "=PyTorch= __TOC__ ==Description== PyTorch is deep learning framework that puts Python first. * Homepage: [https://www.pytorch.org/ https://www.pytorch.org/] ==Access== Pytor...")
 
(PyTorch)
Line 7: Line 7:
 
==Access==
 
==Access==
 
Pytorch is open to all HPRC users.
 
Pytorch is open to all HPRC users.
 +
 +
===Anaconda and Pytorch Packages===
 +
TAMU HPRC currently supports the user of Pytorch though the Anaconda modules. There are a variety of Anaconda modules available on Ada and Terra.
 +
 +
<font color=purple>While several versions of Anaconda have some Pytorch environment installed, it is simplest to use exactly the versions in the following sections.</font>
 +
 +
You can learn more about the module system on our [[SW:Modules]] page.
 +
 +
You can explore the available Anaconda environments on a per-module basis using the following:
 +
[NetID@ada ~]$ '''module load Anaconda/[SomeVersion]'''
 +
[NetID@ada ~]$ '''conda info --envs'''
 +
 +
===Pytorch on Ada (CPU-only)===
 +
A single version of TensorFlow is currently available on Ada. This version is limited to the CPU only (no GPU).
 +
 +
To load this version (python 3.6):
 +
[NetID@ada ~]$ '''module load Anaconda/3-5.0.0.1'''
 +
[NetID@ada ~]$ '''source activate pytorch-0.2.0'''
 +
[NetID@ada ~]$ ''[run your Python program accessing Pytorch]''
 +
[NetID@ada ~]$ '''source deactivate'''
 +
 +
<font color=purple>This version can be run on any of the 64GB or 256GB compute nodes.</font>

Revision as of 16:19, 27 February 2018

PyTorch

Description

PyTorch is deep learning framework that puts Python first.

Access

Pytorch is open to all HPRC users.

Anaconda and Pytorch Packages

TAMU HPRC currently supports the user of Pytorch though the Anaconda modules. There are a variety of Anaconda modules available on Ada and Terra.

While several versions of Anaconda have some Pytorch environment installed, it is simplest to use exactly the versions in the following sections.

You can learn more about the module system on our SW:Modules page.

You can explore the available Anaconda environments on a per-module basis using the following:

[NetID@ada ~]$ module load Anaconda/[SomeVersion]
[NetID@ada ~]$ conda info --envs

Pytorch on Ada (CPU-only)

A single version of TensorFlow is currently available on Ada. This version is limited to the CPU only (no GPU).

To load this version (python 3.6):

[NetID@ada ~]$ module load Anaconda/3-5.0.0.1
[NetID@ada ~]$ source activate pytorch-0.2.0
[NetID@ada ~]$ [run your Python program accessing Pytorch]
[NetID@ada ~]$ source deactivate

This version can be run on any of the 64GB or 256GB compute nodes.