Forcing the device to catch issue #8. Should use CPU.

leaflet757 · Mar 10, 2016 · 29147a3 · 29147a3
1 parent 9e22b7e
commit 29147a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doodle.py b/doodle.py
@@ -58,7 +58,7 @@ class ansi:
 
 # 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', 'device=%s,floatX=float32,print_active_device=False' % (args.device))
+os.environ.setdefault('THEANO_FLAGS', 'device=%s,force_device=True,floatX=float32,print_active_device=False' % (args.device))
 
 # Scientific Libraries
 import numpy as np