PyTorch
=======

The PyTorch `website <https://pytorch.org>`_ describes the library as:

    Tensors and Dynamic neural networks in Python with strong GPU acceleration.

Setup PyTorch on CSD3
---------------------

The python installation will depend on the target partition (CPU or GPU).
To run Pytorch on a CPU partition (cclake, icelake), please do the following:

.. literalinclude:: scripts/pytorch/pytorch_cpu_setup.sh
    :language: bash

To run Pytorch on a GPU partition (ampere), please follow the following instructions:

.. literalinclude:: scripts/pytorch/pytorch_gpu_setup.sh
    :language: bash

	     
		   
Running a PyTorch program
-------------------------
PyTorch will be able to run on both CPU and GPU.

CPU submission script:

.. literalinclude:: scripts/pytorch/pytorch_cpu_sbatch
    :language: bash
	       
GPU submission script:

.. literalinclude:: scripts/pytorch/pytorch_gpu_sbatch
    :language: bash