原因:代码中指定了 特定的GPU,参考代码:

os.environ["CUDA_VISIBLE_DEVICES"] = "6, 7"

解决办法:将这行代码 删除 或者 注释 即可

# os.environ["CUDA_VISIBLE_DEVICES"] = "6, 7" # 不限制GPU可以直接删除这行代码

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐