无法将CUDA_VISIBLE_DEVICES设置为通过Linux终端使用多个GPU进行训练

Question

我正在尝试使用多个GPU进行训练，但不能。我只能使用特定的GPU。

在这里，我将1和2作为GPU ID：

export CUDA_VISIBLE_DEVICES = [1,2]

在使用nvidia-smi进行检查时，我发现仅使用1个GPU。

请告知我如何通过设置CUDA_VISIBLE_DEVICES使用多个GPU。我的GPU ID为0,1,2。

我在Linux终端上尝试的示例代码。

export CUDA_VISIBLE_DEVICES = 0,1,2
export CUDA_VISIBLE_DEVICES = [0,1,2]
export CUDA_VISIBLE_DEVICES = {1,2}
export CUDA_VISIBLE_DEVICES = {0,1,2}
export CUDA_VISIBLE_DEVICES = [1,2]