Difference between revisions of "Infrastructure/software/pytorch"
(→Installation on Abel) |
(→Installation on Abel) |
||
| Line 23: | Line 23: | ||
</pre> | </pre> | ||
| − | Next, we need to create a module | + | Next, we need to create a module definition, in this case |
<tt>/projects/nlpl/software/modulefiles/nlpl-pytorch/0.4.1</tt>. | <tt>/projects/nlpl/software/modulefiles/nlpl-pytorch/0.4.1</tt>. | ||
Revision as of 13:23, 14 September 2018
Background
PyTorch is one of the widely used programming frameworks for Deep Learning; in and of itself it is not NLP-specific, but it is a critical dependency for several of the NLPL strands of activity.
Installation on Abel
The module nlpl-pytorch provides a PyTorch installation in a Python 3.5 virtual environment.
module purge module load cuda/8.0 module load python3/3.5.0
cd /projects/nlpl/software mkdir pytorch virtualenv pytorch/0.4.1
Next, we need to create a module definition, in this case /projects/nlpl/software/modulefiles/nlpl-pytorch/0.4.1.
module load nlpl-pytorch/0.4.1 pip install --upgrade pip pip install --upgrade numpy pillow six pip install torch torchvision
