Remove cnmem theano flag since it doesn't work if you're sharing GPU with display.

2016-10-30 19:24:50 -07:00
parent 5ef872b876
commit fef84c5b44
1 changed files with 1 additions and 1 deletions
@@ -96,7 +96,7 @@ print("""{}   {}Super Resolution for images and videos powered by Deep Learning!
 # Load the underlying deep learning libraries based on the device specified.  If you specify THEANO_FLAGS manually,
 # the code assumes you know what you are doing and they are not overriden!
 os.environ.setdefault('THEANO_FLAGS', 'floatX=float32,device={},force_device=True,allow_gc=True,'\
-                                      'print_active_device=False,lib.cnmem=1.0'.format(args.device))
+                                      'print_active_device=False'.format(args.device))

 # Scientific & Imaging Libraries
 import numpy as np