RuntimeError: No CUDA GPUs are available (python与torch版本匹配，且正确安装cuda版本的torch,但仍然报错)
·
原因：代码中指定了 特定的GPU,参考代码：
os.environ["CUDA_VISIBLE_DEVICES"] = "6, 7"
解决办法：将这行代码 删除 或者 注释 即可
# os.environ["CUDA_VISIBLE_DEVICES"] = "6, 7" # 不限制GPU可以直接删除这行代码
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)