os.environ CUDA_VISIBLE_DEVICES异常问题

代码:

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

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

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

你可能感兴趣的:(异常错误处理,深度学习,pytorch,python)