keras强制使用cpu

安装了gpu版本后,keras预测也使用了GPU。因为项目要求测试CPU下速度,所以,在python调用代码中,作以下限制:python import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
相关文章
相关标签/搜索