How to train on only one GPU?

[carshadi]

Hello,
Which parameters do I need to change to make this train and evaluate on one GPU?
I am currently getting an OOM Resource Exhausted Error when i try to train on one GTX 1080.
I tried setting num_towers=1 in train_imagenet.py but this did not help.

[carshadi]

Also tried reducing the batch size to 32,16,etc but the program got stuck before even starting to iterate

[ArashHosseini]

@carshadi, hi

self-attention-gan/train_imagenet.py

Line 124 in ad9612e

session_config = tf.ConfigProto(

did you try config.gpu_options.allow_growth=True ? or config.gpu_options.per_process_gpu_memory_fraction = 0.4