os.environ[‘CUDA_VISIBLE_DEVICES‘] 无效

eaaaaa 我又要被自己蠢哭了

我搜了好多原因，比如把位置放在torch使用之前，其他文件中重复定义等等原因都没有解决。

后来发现是我敲代码时候用了小写

我写的是： os.environ['cuda_visible_devices'] = "1,2,3"

应该是：    os.environ['CUDA_VISIBLE_DEVICES'] = “1，2，3”

 

最后注释： 该命令作用是定义网络可以看到的物理显卡编号。和

nn.DataParallel(model, device_ids=[0, 1]) 中的编号不同。

这里面是网络可以使用的 os.environ指定的“1，2，3”中的第0，1块。也就是物理的第1，2块。