add pygpu

AirShower_WGAN/ShowerGAN.py

@@ -14,6 +14,11 @@ import tensorflow as tf
 KTF.set_session(utils.get_session())  # Allows 2 jobs per GPU, Please do not change this during the tutorial
 log_dir="."
 
+try:
+    CONDOR_ID = os.environ['CONDOR_ID']
+except:
+    sys.exit('Error: Run this script with "pygpu %file"')
+
 # basic trainings parameter
 EPOCHS = 10
 GRADIENT_PENALTY_WEIGHT = 10

MNIST_GAN/MNIST_GAN.py

@@ -10,6 +10,12 @@ from gan import build_generator, build_discriminator, plot_images, make_trainabl
 
 log_dir="."
 KTF.set_session(get_session())  # Allows 2 jobs per GPU, Please do not change this during the tutorial
+
+try:
+    CONDOR_ID = os.environ['CONDOR_ID']
+except:
+    sys.exit('Error: Run this script with "pygpu %file"')
+
 # prepare MNIST dataset
 data = mnist.load_data()
 X_train = data.train_images.reshape(-1, 28, 28, 1) / 255.