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块。