tensorflow禁用GPU

来源:互联网 发布:淘宝店有流量没销量 编辑:程序博客网 时间:2024/06/16 13:25

如果安装了tensorflow-gpu版,但是在测试时只想用cpu版本,可以作如下修改:

import osos.environ["CUDA_VISIBLE_DEVICES"]="-1"    import tensorflow as tf

参考链接:https://stackoverflow.com/questions/44552585/prevent-tensorflow-from-accessing-the-gpu