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