Page tree

Versions Compared

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

Keywords

Deep Learning, GPU, Python

Description

PyTorch is a popular deep learning framework written in Python with GPU accelation

Version

0.1.12

Official Website

http://pytorch.org/

Licensing Terms and Conditions

https://github.com/pytorch/pytorch/blob/master/LICENSE

Usage

Example GPU Job Script

Use this template and replace the pytorch loading part (e.g., activate your conda env or from pip). It should return 1.

Code Block
themeMidnight
#!/bin/bash
#SBATCH -p serial
#SBATCH -C gpu
# Walltime format hh:mm:ss
#SBATCH --ntasks=1
#SBATCH --time=48:00:00
# Output and error files
#SBATCH -o job.%J.out
#SBATCH -e job.%J.err
# **** Put all #SBATCH directives above this line! ****
# **** Otherwise they will not be in effective! ****
#
# **** Actual commands start here ****
# Load modules here (safety measure)
module purge
# Activate pytorch here
module load gcc cuda
echo "Replace this line with loading the Python env with pytorch here"; exit 1
# Actual command here
python -m 'import torch; print(torch.cuda.is_available())'

 

FAQ

 

Panel

On This Page:

Table of Contents
maxLevel5
Panel

Software List

Content Report Table
maxResults1000
spacesADRC
labelshpc-software