文章讲述了在使用PyTorch时，尽管配置文件显示设备为7，但程序实际使用的是0号卡。解决方法是在Python脚本中设置CUDA_VISIBLE_DEVICES环境变量，以指定设备使用顺序。作者通过实验发现设备顺序对程序运行的实际GPU选择有影响。
错误描述：
在配置文件以及输出日志明确可以看到device=7，但程序仍跑在0号卡。
解决办法：
CUDA_VISIBLE_DEVICES=6,7 python train-seg.py #在python语句前加这句相当于程序只看得见6，7号卡，自然不会用其他号卡了。但是需要注意此时卡的顺序需要改变的。
经过我的实验，发现，若以以上为准，指定device=1，那么程序将在6号卡运行，若CUDA_VISIBLE_DEVICES=5,6,7，此时指定device=1，程序将在5号卡运行，device=2，将在6号卡运行。
参考文章：为什么pytorch死活要用0号卡
2442
1833
2106
552
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
