CUDA_VISIBLE_DEVICES是什么?
时间: 2024-04-04 22:25:44 浏览: 117
SGE_GPU:补丁到 SGE 2011.p1,网格引擎之子,多 GPU,Linux-开源
CUDA_VISIBLE_DEVICES是一个环境变量,用于指定程序可见的GPU设备编号。通过设置这个环境变量,我们可以在多GPU服务器上选择特定的GPU设备来运行程序。例如,当我们设置CUDA_VISIBLE_DEVICES=1时,程序只能看到服务器上的第二个GPU设备。这个环境变量在深度学习等需要GPU加速的任务中非常有用。
阅读全文