os.environ["cuda_visible_devices"]

os.environ["cuda_visible_devices"] 是在 Python 中使用 CUDA 时的一个环境变量，它用于指定在程序中可见的 CUDA 设备。其中的值可以是一个整数（如 "0"），表示只使用编号为 0 的 CUDA 设备，也可以是一个由逗号分隔的整数列表（如 "0,1,2"），表示使用编号为 0、1、2 的 CUDA 设备。