代码:
os.environ[“CUDA_VISIBLE_DEVICES”] = “0,1,2,3,4,5,6,7”
系统提示只发现了 gpu:0,参考网络资料,检查发现有一个调用的py文件里代码是:os.environ[“CUDA_VISIBLE_DEVICES”] = “0”
删掉该代码,发现全部的GPU进行调用。
代码:
os.environ[“CUDA_VISIBLE_DEVICES”] = “0,1,2,3,4,5,6,7”
系统提示只发现了 gpu:0,参考网络资料,检查发现有一个调用的py文件里代码是:os.environ[“CUDA_VISIBLE_DEVICES”] = “0”
删掉该代码,发现全部的GPU进行调用。