Tensorflow强制使用CPU
对于Tensorflow,声明Session的时候加入device_count={‘cpu’:0}即可,代码如下:
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(device_count={'cpu':0}));
第2种是使用CUDA_VISIBLE_DEVICES命令行参数,代码如下:
CUDA_VISIBLE_DEVICES="" python3 train.py
网上还有说直接改环境变量的
第2种是使用CUDA_VISIBLE_DEVICES命令行参数,代码如下:
CUDA_VISIBLE_DEVICES=""
这个还没尝试