Python GPU Programming

How to use Python with GPU Support

Multiple libraries are available which allow utilizing GPU with Python. Currently on RAAD we have below packages available which can utilize GPU functionality with Python.

1. GPU accelerated Computing with Python
2. PyCuda
3. Deep Learning using cuDNN
4. Theano

How to get started?

Submit Interactive or Batch job in Graphics queue on RAAD.

#PBS -S /bin/bash
#PBS -q graphics
#PBS -N GPU_Job
#PBS -l select=1:ncpus=16:mem=62Gb
#PBS -l walltime=8:00:00

module load python/2.7.9
module load cuda/7.5

cd $PBS_O_WORKDIR
python2.7 <your_Script>