Skip to content

Instantly share code, notes, and snippets.

@enewe101

enewe101/gpu-helios.sh

Created October 26, 2016 16:40
Show Gist options
  • Save enewe101/a4ac07f1aeb34afce001766c58fa5062 to your computer and use it in GitHub Desktop.
Save enewe101/a4ac07f1aeb34afce001766c58fa5062 to your computer and use it in GitHub Desktop.
Download ZIP
Run Theano GPU code on Helios
Raw
gpu-helios.sh
#!/bin/bash
#PBS -l walltime=12:00:00
#PBS -l nodes=1:gpus=1
#PBS -A eeg-641-aa
#PBS -e /home/enewel3/jobs/relation2vec/proto.stderr
#PBS -o /home/enewel3/jobs/relation2vec/proto.stdout
#PBS -N r2v-proto
# Load modules, source virtualenv, go to source dir
module load apps/python/2.7.10 cuda/7.5.18 libs/cuDNN/4
source /home/enewel3/entity-embeddings/venv/bin/activate
cd /home/enewel3/entity-embeddings/src/embedder
# Run the model
THEANO_FLAGS='floatX=float32,device=gpu' ./train_r2v.py
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment