Backto TensorFlow Index.
import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
import tensorflow as tf
原理:TF 读取 GPU cards,是从 0 开始编号的。所以把GPU数量设置为 -1,TF 就乖乖的用 CPU 了。
Backto TensorFlow Index.
import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"
import tensorflow as tf
原理:TF 读取 GPU cards,是从 0 开始编号的。所以把GPU数量设置为 -1,TF 就乖乖的用 CPU 了。