Tensorflow强制使用CPU

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="" 

这个还没尝试

你可能感兴趣的:(tensorflow,程序)