如果安装了tensorflow-gpu版,但是在测试时只想用cpu版本,可以作如下修改:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
import tensorflow as tf
如果安装了tensorflow-gpu版,但是在测试时只想用cpu版本,可以作如下修改:
import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
import tensorflow as tf