The config module contains many attributes that modify Theano’s behavior. Many of these attributes are consulted during the import of the theano module and many are assumed to be read-only.
As a rule, the attributes in this module should not be modified by user code.
Theano’s code comes with default values for these attributes, but you can override them from your .theanorc file, and override those values in turn by the THEANO_FLAGS environment variable.
The order of precedence is:
You can print out the current/effective configuration at any time by printing theano.config. For example, to see a list of all active configuration variables, type this from the command-line:
python -c 'import theano; print theano.config' | less
This is a list of comma-delimited key=value pairs that control Theano’s behavior.
For example, in bash, you can override your THEANORC defaults for <myscript>.py by typing this:
THEANO_FLAGS='floatX=float32,device=gpu0,nvcc.fastmath=True' python <myscript>.py
If a value is defined several times in THEANO_FLAGS, the right-most definition is used. So, for instance, if THEANO_FLAGS='device=cpu,device=gpu0', then gpu0 will be used.
The location[s] of the .theanorc file[s] in ConfigParser format. It defaults to $HOME/.theanorc. On Windows, it defaults to $HOME/.theanorc:$HOME/.theanorc.txt to make Windows users’ life easier.
Here is the .theanorc equivalent to the THEANO_FLAGS in the example above:
[global]
floatX = float32
device = gpu0
[nvcc]
fastmath = True
Configuration attributes that are available directly in config (e.g. config.device, config.mode) should be defined in the [global] section. Attributes from a subsection of config (e.g. config.nvcc.fastmath, config.blas.ldflags) should be defined in their corresponding section (e.g. [nvcc], [blas]).
Multiple configuration files can be specified by separating them with ‘:’ characters (as in $PATH). Multiple configuration files will be merged, with later (right-most) files taking priority over earlier files in the case that multiple files specify values for a common configuration option. For example, to override system-wide settings with personal ones, set THEANORC=/etc/theanorc:~/.theanorc.
The list below describes some of the more common and important flags that you might want to use. For the complete list (including documentation), import theano and print the config variable, as in:
python -c 'import theano; print theano.config' | less
String value: either 'cpu', 'gpu', 'gpu0', 'gpu1', 'gpu2', or 'gpu3'
Default device for computations. If gpu*, change the default to try to move computation to it and to put shared variable of float32 on it. Choose the default compute device for theano graphs. Setting this to a gpu* string will make theano to try by default to move computation to it. Also it will make theano put by default shared variable of float32 on it. 'gpu' lets the driver select the GPU to use, while 'gpu?' makes Theano try to use a specific device. If we are not able to use the GPU, either we fall back on the CPU, or an error is raised, depending on the force_device flag.
This flag’s value cannot be modified during the program execution.
Do not use upper case letters, only lower case even if NVIDIA use capital letters.
Bool value: either True or False
Default: False
If True and device=gpu*, we raise an error if we cannot use the specified device. If True and device=cpu, we disable the GPU. If False and device=gpu*, and if the specified device cannot be used, we warn and fall back to the CPU.
This is useful to run Theano’s tests on a computer with a GPU, but without running the GPU tests.
This flag’s value cannot be modified during the program execution.
String value: either '', 'gpu', 'gpu0', 'gpu1', 'gpu2', or 'gpu3'
Initialize the gpu device to use. When its value is gpu*, the theano flag device must be "cpu". Unlike device, setting this flag to a specific GPU will not try to use this device by default, in particular it will not move computations, nor shared variables, to the specified GPU.
This flag is useful to run GPU-specific tests on a particular GPU, instead of using the default one.
This flag’s value cannot be modified during the program execution.
Bool value: either True or False
Default: False
If True, always initialize PyCUDA when Theano want to initialize the GPU. With PyCUDA version 2011.2.2 or earlier, PyCUDA must initialize the GPU before Theano does it. Setting this flag to True, ensure that, but always import PyCUDA. It can be done manually by importing theano.misc.pycuda_init before Theano initialize the GPU device. Newer version of PyCUDA (currently only in the trunk) don’t have this restriction.
Bool value: either True or False
Default: True
Print active device at when the GPU device is initialized.
String value: either ‘float64’ or ‘float32’
Default: ‘float64’
This sets the default dtype returned by tensor.matrix(), tensor.vector(), and similar functions. It also sets the default theano bit width for arguments passed as Python floating-point numbers.
String value: either ‘ignore’, ‘warn’, ‘raise’ or ‘pdb’
Default: ‘ignore’
When creating a TensorVariable with dtype float64, what should be done? This is useful to help find upcast to float64 in user code.
Bool value: either True or False
Default: True
This sets the default for the use of the Theano garbage collector for intermediate results. To use less memory, Theano frees the intermediate results as soon as they are no longer needed. Disabling Theano garbage collection allows Theano to reuse buffers for intermediate results between function calls. This speeds up Theano by no longer spending time reallocating space. This gives significant speed up on functions with many ops that are fast to execute, but this increases Theano’s memory usage.
Bool value: either True or False
Enable or not parallel computation on the CPU with OpenMP. It is the default value used when creating an Op that support it. The best is to define it via Theano configuration file or with the environment variable THEANO_FLAGS.
Positive int value, default: 200000.
This specifies the vectors minimum size for which elemwise ops use openmp, if openmp is enabled.
String value: either ‘numpy+floatX’ or ‘custom’
Default: ‘custom’
This specifies how data types are implicitly figured out in Theano, e.g. for constants or in the results of arithmetic operations. The ‘custom’ value corresponds to a set of custom rules originally used in Theano (which can be partially customized, see e.g. the in-code help of tensor.NumpyAutocaster), and will be deprecated in the future. The ‘numpy+floatX’ setting attempts to mimic the numpy casting rules, although it prefers to use float32 numbers instead of float64 when config.floatX is set to ‘float32’ and the user uses data that is not explicitly typed as float64 (e.g. regular Python floats). Note that ‘numpy+floatX’ is not currently behaving exactly as planned (it is a work-in-progress), and thus you should consider it as experimental. At the moment it behaves differently from numpy in the following situations:
String value: either ‘int’, ‘floatX’ or ‘raise’
Default: ‘int’
Specifies what to do when one tries to compute x / y, where both x and y are of integer types (possibly unsigned). ‘int’ means an integer is returned (as in Python 2.X), but this behavior is deprecated. ‘floatX’ returns a number of type given by config.floatX. ‘raise’ is the safest choice (and will become default in a future release of Theano) and raises an error when one tries to do such an operation, enforcing the use of the integer division operator (//) (if a float result is intended, either cast one of the arguments to a float, or use x.__truediv__(y)).
String value: ‘Mode’, ‘ProfileMode’(deprecated), ‘DebugMode’, ‘FAST_RUN’, ‘FAST_COMPILE’
Default ‘Mode’
This sets the default compilation mode for theano functions. By default the mode Mode is equivalent to FAST_RUN. See Config attribute linker and optimizer.
Bool value: either True or False
Default False
Do the vm/cvm linkers profile the execution time of Theano functions?
See Profiling Theano function for examples.
Bool value: either True or False
Default False
Do the vm/cvm linkers profile the memory usage of Theano functions? It only works when profile=True.
Bool value: either True or False
Default False
Do the vm/cvm linkers profile the optimization phase when compiling a Theano function? It only works when profile=True.
Positive int value, default: 20.
The number of Apply nodes to print in the profiler output
Positive int value, default: 20.
The number of Ops to print in the profiler output
Positive int value, default: 1024.
For the memory profile, do not print Apply nodes if the size of their outputs (in bytes) is lower than this.
Bool value: either True or False
Default False
Does the memory profile print the min peak memory usage? It only works when profile=True, profile_memory=True
String value: ‘stderr’, ‘stdout’, or a name of a file to be created
Default ‘stderr’
Name of the destination file for the profiling output. The profiling output can be either directed to stderr (default), or stdout or an arbitrary file.
Bool value: either True or False
Default False
This makes the compilation use the amdlibm library, which is faster than the standard libm.
String value: ‘c|py’, ‘py’, ‘c’, ‘c|py_nogc’
Default: ‘c|py’
When the mode is Mode, it sets the default linker used. See Configuration Settings and Compiling Modes for a comparison of the different linkers.
String value: ‘fast_run’, ‘merge’, ‘fast_compile’, ‘None’
Default: ‘fast_run’
When the mode is Mode, it sets the default optimizer used.
String value: ‘warn’, ‘raise’ or ‘pdb’
Default: ‘warn’
When a crash occurs while trying to apply some optimization, either warn the user and skip this optimization (‘warn’), raise the exception (‘raise’), or fall into the pdb debugger (‘pdb’).
String value: ‘warn’ or ‘raise’
Default: ‘warn’
When an exception is raised when inferring the shape of some apply node, either warn the user and use a default value (‘warn’), or raise the exception (‘raise’).
String value: ‘None’, ‘all’, ‘0.3’, ‘0.4’, ‘0.4.1’, ‘0.5’, ‘0.6’, ‘0.7’
Default: ‘0.6’
When we fix a Theano bug that generated bad results under some circumstances, we also make Theano raise a warning when it encounters the same circumstances again. This helps to detect if said bug had affected your past experiments, as you only need to run your experiment again with the new version, and you do not have to understand the Theano internal that triggered the bug. A better way to detect this will be implemented. See this ticket.
This flag allows new users not to get warnings about old bugs, that were fixed before their first checkout of Theano. You can set its value to the first version of Theano that you used (probably 0.3 or higher)
None means that all warnings will be displayed. all means all warnings will be ignored.
It is recommended that you put a version, so that you will see future warnings. It is also recommended you put this into your .theanorc, so this setting will always be used.
This flag’s value cannot be modified during the program execution.
Default: On Windows: $LOCALAPPDATA\Theano if $LOCALAPPDATA is defined, otherwise and on other systems: ~/.theano.
This directory stores the platform-dependent compilation directories.
This flag’s value cannot be modified during the program execution.
Default: “compiledir_%(platform)s-%(processor)s-%(python_version)s-%(python_bitwidth)s”
This is a Python format string that specifies the subdirectory of config.base_compiledir in which to store platform-dependent compiled modules. To see a list of all available substitution keys, run python -c "import theano; print theano.config", and look for compiledir_format.
This flag’s value cannot be modified during the program execution.
Default: config.base_compiledir/config.compiledir_format
This directory stores dynamically-compiled modules for a particular platform.
This flag’s value cannot be modified during the program execution.
Default: ‘-lblas’
Link arguments to link against a (Fortran) level-3 blas implementation. The default will test if ‘-lblas’ work. If not, we will disable our c code for BLAS.
Bool value: either True or False
Default: True
When the local_alloc_optimization is applied, add an assert to highlight shape errors.
Without such asserts this optimization could hide errors in the user code. We add the assert only if we can’t infer that the shapes are equivalent. As such this optimization does not always introduce an assert in the graph. Removing the assert could speed up execution.
Default: $CUDA_ROOT or failing that, “/usr/local/cuda”
A directory with bin/, lib/, include/ folders containing cuda utilities.
String value: ‘none’, ‘small’, ‘large’
Default: ‘small’
The default value for the amount of working memory that is tolerated in the convolution implementation in cudnn.
Default: “”
Extra parameters to pass to gcc when compiling. Extra include paths, library paths, configuration options, etc.
Default: Full path to g++ if g++ is present. Empty string otherwise.
Indicates which C++ compiler to use. If empty, no C++ code is compiled. Theano automatically detects whether g++ is present and disables C++ compilation when it is not. On darwin systems (Mac OS X), it preferably looks for clang++ and uses that if available.
We print a warning if we detect that no compiler is present. It is recommended to run with C++ compilation as Theano will be much slower otherwise.
This can be any compiler binary (full path or not) but things may break if the interface is not g++-compatible to some degree.
Default: False
If true, this will enable fastmath (–use_fast_math) mode for compiled cuda code which makes div and sqrt faster at the cost of precision. This also disables support for denormal numbers.
Default: “”
A list of optimizer tags that we don’t want included in the default Mode. If multiple tags, separate them by ‘:’. Ex: to remove the elemwise inplace optimizer(slow for big graph), use the flags: optimizer_excluding:inplace_opt, where inplace_opt is the name of that optimization.
This flag’s value cannot be modified during the program execution.
Default: “”
A list of optimizer tags that we want included in the default Mode. If multiple tags, separate them by ‘:’.
This flag’s value cannot be modified during the program execution.
Default: “”
A list of optimizer tags that we require for optimizer in the default Mode. If multiple tags, separate them by ‘:’.
This flag’s value cannot be modified during the program execution.
Bool value: either True or False
Default: False
When True, we print on the stdout the optimization applied.
Bool value: either True or False
Default: False
If False, source code files are removed when they are not needed anymore. This means files whose compilation failed are deleted. Set to True to keep those files in order to debug compilation errors.
This section contains attributes which influence the compilation of C code for ops. Due to historical reasons many attributes outside of this section also have an influence over compilation, most notably ‘cxx’. This is not expected to change any time soon.
Positive int value, default: compile.wait * 24
Time to wait before an unrefreshed lock is broken and stolen. This is in place to avoid manual cleanup of locks in case a process crashed and left a lock in place.
The refresh time is automatically set to half the timeout value.
Positive int value, default: 5
Time to wait between attempts at grabbing the lock if the first attempt is not successful. The actual time will be between compile.wait and compile.wait * 2 to avoid a crowding effect on lock.
This section contains various attributes configuring the behaviour of mode DebugMode. See directly this section for the documentation of more configuration options.
Default: ''
A list of kinds of preallocated memory to use as output buffers for each Op’s computations, separated by :. Implemented modes are:
In order not to test with preallocated memory, use an empty string, "".
Positive int value, default: 4.
When testing with “strided” preallocated output memory, test all combinations of strides over that number of (inner-most) dimensions. You may want to reduce that number to reduce memory or time usage, but it is advised to keep a minimum of 2.
Bool value, default: True
Generate a warning when the destroy_map or view_map tell that an op work inplace, but the op did not reuse the input for its output.
This section contains different attributes for configuring numpy’s behaviour, described by numpy.seterr.
String Value: 'ignore', 'warn', 'raise', 'call', 'print', 'log', 'None'
Default: 'ignore'
Set the default behaviour described by numpy.seterr.
'None' means that numpy’s default behaviour will not be changed (unless one of the other config.numpy.seterr_* overrides it), but this behaviour can change between numpy releases.
This flag sets the default behaviour for all kinds of floating-pont errors, and it can be overriden for specific errors by setting one (or more) of the flags below.
This flag’s value cannot be modified during the program execution.
String Value: 'None', 'ignore', 'warn', 'raise', 'call', 'print', 'log'
Default: 'None'
Sets numpy’s behavior for division by zero. 'None' means using the default, defined by config.numpy.seterr_all.
This flag’s value cannot be modified during the program execution.
String Value: 'None', 'ignore', 'warn', 'raise', 'call', 'print', 'log'
Default: 'None'
Sets numpy’s behavior for floating-point overflow. 'None' means using the default, defined by config.numpy.seterr_all.
This flag’s value cannot be modified during the program execution.
String Value: 'None', 'ignore', 'warn', 'raise', 'call', 'print', 'log'
Default: 'None'
Sets numpy’s behavior for floating-point underflow. 'None' means using the default, defined by config.numpy.seterr_all.
This flag’s value cannot be modified during the program execution.
String Value: 'None', 'ignore', 'warn', 'raise', 'call', 'print', 'log'
Default: 'None'
Sets numpy’s behavior for invalid floating-point operation. 'None' means using the default, defined by config.numpy.seterr_all.
This flag’s value cannot be modified during the program execution.
String Value: 'off', 'ignore', 'warn', 'raise'.
Default: 'off'
Setting this attribute to something other than 'off' activates a debugging mechanism, where Theano executes the graph on-the-fly, as it is being built. This allows the user to spot errors early on (such as dimension mis-match), before optimizations are applied.
Theano will execute the graph using the Constants and/or shared variables provided by the user. Purely symbolic variables (e.g. x = T.dmatrix()) can be augmented with test values, by writing to their 'tag.test_value' attribute (e.g. x.tag.test_value = numpy.random.rand(5,4)).
When not 'off', the value of this option dictates what happens when an Op’s inputs do not provide appropriate test values:
- 'ignore' will silently skip the debug mechanism for this Op
- 'warn' will raise a UserWarning and skip the debug mechanism for this Op
- 'raise' will raise an Exception
As compute_test_value, but it is the value used during Theano optimization phase. Theano user’s do not need to use this. This is to help debug shape error in Theano optimization.
Bool value, default: True
Theano users can use the standard python pickle tools to save a compiled theano function. When pickling, both graph before and after the optimization are saved, including shared variables. When set to True, the graph is reoptimized when being unpickled. Otherwise, skip the graph optimization and use directly the optimized graph.
String Value: 'low', 'high'.
Default: 'low'
If 'low', the text of exceptions will generally refer to apply nodes with short names such as 'Elemwise{add_no_inplace}'. If 'high', some exceptions will also refer to apply nodes with long descriptions like:
A. Elemwise{add_no_inplace}
B. log_likelihood_v_given_h
C. log_likelihood_h
Bool value, default: False
If True, will print a warning when compiling one or more Op with C code that can’t be cached because there is no c_code_cache_version() function associated to at least one of those Ops.
Bool value, default: False
If set to True, breaks certain MacOS installations with the infamous Bus Error.
Bool value, default: False
If True, will remove the -O* parameter passed to g++. This is useful to debug in gdb modules compiled by Theano. The parameter -g is passed by default to g++.
Bool value, default: False
If True, will print compilation warnings.
Bool value, default: False
If set to True, will preload the C module cache at import time