How to specify the gpu device for cuda backend?

[MuGdxy]

How can I manually specify the Device ID to select the GPU when using the Cuda backend?

[MuGdxy]

You have two ways to achieve that:

override the engine config or modify it in code.

A. Override Engine Default Config Externally

#57

1. Create a file named uipc_engine_config.json with content:

   {
       "gpu": 
       {
           "device": 0 // set by user
       }
   }
   

2. Set the Environment Variable UIPC_ENGINE_DEFAULT_CONFIG=<CONFIG_FILE>. <CONFIG_FILE> is the full path of uipc_engine_config.json, e.g. /home/my_name/uipc_engine_config.json.

This approach doesn't require any code changing. You only need to create a config file for a computer.

B. Specify in Code

Approach A overrides the uipc engine's default config. In contrast, Approach B modifies the config JSON in code.

config = Engine.default_config() // Approach A happens here
config['gpu']['device'] = 0 // Approach B happens here
engine = Engine('cuda', config)