【踩坑】PyTorch中指定GPU不生效和GPU编号不一致问题
原创【踩坑】PyTorch中指定GPU不生效和GPU编号不一致问题
原创
小锋学长生活大爆炸
发布于 2024-04-26 04:20:37
发布于 2024-04-26 04:20:37
转载请注明出处：小锋学长生活大爆炸[xfxuezhang.cn]
指定GPU不生效问题
解释：就是使用os.environ["CUDA_VISIBLE_DEVICES"] = "1"后，后面使用起来仍然是cuda0.
解决：在最开头就使用
import osos.environ["CUDA_VISIBLE_DEVICES"] = "0,2" # xxxxxx其它代码不过需要注意的是，这样设置之后，GPU的编号会变
GPU编号不一致问题
解释：torch.device("cuda:0")所选择的GPU跟nvidia-smi中的GPU编号不一样
解决：指定编号方式
import osos.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" # xxxxxx其它代码
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。
如有侵权，请联系 cloudcommunity@tencent.com 删除。
评论
登录后参与评论
推荐阅读
目录