当前位置: 代码迷 >> 综合 >> tensorflow2 GPU和CPU切换
  详细解决方案

tensorflow2 GPU和CPU切换

热度:41   发布时间:2024-02-19 09:11:42.0

当你安装上gpu版本的tensorflow2时,默认情况下使用GPU,当你想要切换回CPU运行程序时,

你需要的代码开头添加一句代码

os.environ["CUDA_VISIBLE_DEVICES"] = "-1"