Commit 3a6d7c77 authored by Gao, Shang's avatar Gao, Shang
Browse files

fixed pbs to run on rhea

parent a1595128
Loading
Loading
Loading
Loading
+4 −1
Original line number Diff line number Diff line
@@ -7,6 +7,9 @@

module load cudatoolkit
module load cuda
module unload PE-intel/14.0.4-1.8.4
module load PE-gnu/4.8.2-1.8.4
module load gcc/4.8.2

#clone repo/get latest
cd ${HOME}/
@@ -28,7 +31,7 @@ if [ ! -f 1FME-0_contact_matrix.npy ]; then
    python dcd2array.py ${HOME}/1fme-0 1:29
fi

python conv_vae.py 1FME-0_contact_matrix.npy
THEANO_FLAGS=device=gpu,floatX=float32,dnn.enabled=False,'gcc.cxxflags=-march=core2' python conv_vae.py 1FME-0_contact_matrix.npy

#cleanup
cd ${HOME}/
+1 −1
Original line number Diff line number Diff line
@@ -38,7 +38,7 @@ for resource in resources:
                               
#wait for downloads to complete
print "waiting for downloads to complete"
timeout = 600
timeout = 200
start = time.time()
for resource in dl:
    while not cbow.check_resource_exists_endpoint('/~/1fme-0/%s' % resource):