Skip to main content
Version: v1.1.3

Global Settings

Backends

  • To specify which kind of architecture (Arch) to use: ti.init(arch=ti.cuda).
  • To specify the pre-allocated memory size for CUDA: ti.init(device_memory_GB=0.5).
  • To specify which GPU to use for CUDA: export CUDA_VISIBLE_DEVICES=[gpuid].
  • To specify which GPU to use for VULKAN:
  • export TI_VISIBLE_DEVICE=[gpuid]
  • To disable a backend (CUDA, METAL, OPENGL) on start up, e.g. CUDA: export TI_ENABLE_CUDA=0.
note

In case you want to use taichi cuda backend together with GGUI on a machine with multiple GPU cards, please make sure CUDA_VISIBLE_DEVICES matches TI_VISIBLE_DEVICE if any of them exists. In general, CUDA_VISIBLE_DEVICES and TI_VISIBLE_DEVICE should point to a GPU device with the same UUID. Use nvidia-smi -L to query the details of your GPU devices.

Compilation

  • Disable advanced optimization to save compile time & possible errors: ti.init(advanced_optimization=False).
  • Disable fast math to prevent possible undefined math behavior: ti.init(fast_math=False).
  • To print intermediate IR generated: ti.init(print_ir=True). Note that compiled kernels are cached by default. To force compilation and IR emission, use ti.init(print_ir=True, offline_cache=False).

Runtime

  • Restart the entire Taichi system (destroy all fields and kernels): ti.reset().
  • To start program in debug mode: ti.init(debug=True) or ti debug your_script.py.
  • To disable importing torch on start up: export TI_ENABLE_TORCH=0.
  • To disable importing paddle on start up: export TI_ENABLE_PADDLE=0.
  • To set a custom seed for the random number generator used by ti.random(): ti.init(random_seed=seed) where seed is an integer. For example ti.init(random_seed=int(time.time())).

Logging

  • Show more detailed log to level TRACE: ti.init(log_level=ti.TRACE) or ti.set_logging_level(ti.TRACE).
  • Eliminate verbose outputs: ti.init(verbose=False).

Develop

  • To trigger GDB when Taichi crashes: ti.init(gdb_trigger=True).
  • Cache compiled runtime bitcode in dev mode to save start up time: export TI_CACHE_RUNTIME_BITCODE=1.
  • To specify how many threads to run test: export TI_TEST_THREADS=4 or python tests/run_tests.py -t4.

Specifying ti.init arguments from environment variables

Arguments for ti.init may also be specified from environment variables. For example:

  • ti.init(arch=ti.cuda) is equivalent to export TI_ARCH=cuda.
  • ti.init(log_level=ti.TRACE) is equivalent to export TI_LOG_LEVEL=trace.
  • ti.init(debug=True) is equivalent to export TI_DEBUG=1.

If both ti.init argument and the corresponding environment variable are specified, then the one in the environment variable will override the one in the argument, e.g.:

  • if ti.init(arch=ti.cuda) and export TI_ARCH=opengl are specified at the same time, then Taichi will choose ti.opengl as backend.
  • if ti.init(debug=True) and export TI_DEBUG=0 are specified at the same time, then Taichi will disable debug mode.

If ti.init is called twice, then the configuration in first invocation will be completely discarded, e.g.:

ti.init(debug=True)
print(ti.cfg.debug) # True
ti.init()
print(ti.cfg.debug) # False

:::

Was this helpful?