pytorch使用CUDA_VISIBLE_DEVICES注意事项

如果使用了CUDA_VISIBLE_DEVICES=0(或者其它显卡id)，也就是仅一张显卡可见时，代码中的device必须设置为"cuda:0"。同理当设置两张显卡可见时，device最多设置为"cuda:1"，以此类推。