原因:代码中指定了 特定的GPU,参考代码:
os.environ["CUDA_VISIBLE_DEVICES"] = "6, 7"
解决办法:将这行代码 删除 或者 注释 即可
# os.environ["CUDA_VISIBLE_DEVICES"] = "6, 7" # 不限制GPU可以直接删除这行代码
原因:代码中指定了 特定的GPU,参考代码:
os.environ["CUDA_VISIBLE_DEVICES"] = "6, 7"
解决办法:将这行代码 删除 或者 注释 即可
# os.environ["CUDA_VISIBLE_DEVICES"] = "6, 7" # 不限制GPU可以直接删除这行代码