Tensorflow强制使用CPU

Tensorflow强制使用CPU

对于Tensorflow，声明Session的时候加入device_count={‘gpu’:0}即可，代码如下：

import tensorflow as tf 
sess = tf.Session(config=tf.ConfigProto(device_count={'gpu':0}));

第２种是使用CUDA_VISIBLE_DEVICES命令行参数，代码如下：

CUDA_VISIBLE_DEVICES="" python3 train.py

网上还有说直接改环境变量的
第２种是使用CUDA_VISIBLE_DEVICES命令行参数，代码如下：

CUDA_VISIBLE_DEVICES="" 

这个还没尝试