Changing torch_extensions path

Hi guys,

Is there a way to change the default extension path from ~/.cache/torch_extensions/ to something else?
I couldn’t find anything on the documentation.

Thank you,

You should be able to set this path via TORCH_EXTENSIONS_DIR. If this env variable is not set, the (system) default will be used.

1 Like