tensorflow 遇到这个问题,很有可能原因是:
在多卡的机器上,别人占用了其中几个GPU,而你在跑代码的时候,把别人占用的GPU也读进去了。
在全局设置一下你需要用到的GPU即可。
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '2, 3' #use GPU with ID=2, 3
本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!