调用nvrtc.nvrtcCompileProgram后如何恢复locale.getpreferredencoding默认值？

一、首选区域设置 vs 默认区域设置 #

• 默认区域设置：由系统环境变量（如LANG、LC_ALL等）决定，是Python启动时从系统继承的全局区域配置，可通过locale.getdefaultlocale()获取。它影响日期格式、数字格式化等大部分系统级本地化操作。
• 首选编码：由locale.getpreferredencoding()返回，是Python根据当前进程的区域设置状态，推断出的文本I/O（比如open()函数默认编码）专用编码。它不直接绑定系统环境变量——哪怕环境变量保持不变，进程内区域设置被修改后，该值也会同步变化。

你遇到的情况就是nvrtc调用底层修改了进程的区域设置状态，但系统环境变量未变动，导致getpreferredencoding()返回值从UTF-8变为ANSI_X3.4-1968。

二、恢复首选编码至UTF-8的方法 #

常规locale函数可能无法直接修正nvrtc带来的变动，可尝试以下几种方案：

1. 显式指定文件I/O编码 #

最直接的临时规避方案：所有open()调用都显式声明encoding="utf-8"，彻底脱离默认编码依赖：

with open("target_file.txt", "r", encoding="utf-8") as f:
    content = f.read()

2. 强制重置进程区域设置 #

getpreferredencoding()的返回值与LC_CTYPE区域设置强相关，可尝试强制重置该配置：

import locale

# 替换为你系统支持的UTF-8 locale，比如zh_CN.UTF-8
locale.setlocale(locale.LC_CTYPE, "en_US.UTF-8")

若出现locale不存在的报错，可先修改环境变量再重新加载：

import os
import locale

os.environ["LC_CTYPE"] = "UTF-8"
os.environ["LANG"] = "en_US.UTF-8"
locale.textdomain("messages")  # 触发区域设置重新初始化

3. 封装nvrtc调用，前后做状态保存与恢复 #

在调用nvrtcCompileProgram前后，保存当前区域设置状态，调用完成后立即恢复：

import locale
import nvrtc

# 保存当前LC_CTYPE配置
original_locale = locale.getlocale(locale.LC_CTYPE)

# 执行nvrtc编译逻辑
prog = nvrtc.nvrtcCreateProgram(...)
nvrtc.nvrtcCompileProgram(prog, ...)

# 恢复原有区域设置
locale.setlocale(locale.LC_CTYPE, original_locale)

这种方法能从根源上隔离nvrtc对后续代码的影响。

4. 猴子补丁强制锁定首选编码 #

如果上述方法都无效，可通过猴子补丁直接覆盖locale.getpreferredencoding的返回值：

import locale

original_get_encoding = locale.getpreferredencoding

def force_utf8(*args, **kwargs):
    return "UTF-8"

locale.getpreferredencoding = force_utf8

该方法较为直接，适合其他方案均无法解决的极端场景。

内容的提问来源于stack exchange，提问作者Redsnic