0
点赞
收藏
分享

微信扫一扫

os.environ CUDA_VISIBLE_DEVICES异常问题

IT影子 2022-02-02 阅读 40

代码:

os.environ[“CUDA_VISIBLE_DEVICES”] = “0,1,2,3,4,5,6,7”

系统提示只发现了 gpu:0,参考网络资料,检查发现有一个调用的py文件里代码是:os.environ[“CUDA_VISIBLE_DEVICES”] = “0”

删掉该代码,发现全部的GPU进行调用。

举报

相关推荐

0 条评论