选择 GPU / CPU

要在特定 GPU 上运行程序，可以使用 CUDA_VISIBLE_DEVICES 环境变量。例如，若要在第二个 GPU 上运行程序，可以使用：

CUDA_VISIBLE_DEVICES=1 python my_program.py

要在多个 GPU 上运行程序，可以使用：

CUDA_VISIBLE_DEVICES=0,1 python my_program.py

要禁用 GPU（即使用 CPU），可以使用：

CUDA_VISIBLE_DEVICES="" python my_program.py