Running BERT on CPU instead of GPU

Question

I am trying to execute BERT's run_clasifier.py script using terminal as below:

python run_classifier.py --task_name=cola --do_predict=true --data_dir= --vocab_file=$BERT_BASE_DIR/vocab.txt --bert_config_file=$BERT_BASE_DIR/bert_config.json --init_checkpoint=/model.ckpt-1603 --max_seq_length=512 --output_dir=

This by default executes on GPU. Instead, I want to execute it on the CPU. Is there a way to do it for only a single trial of execution than for all the subsequent trials.?

user_19637836 · Answer

Try use: os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

Running BERT on CPU instead of GPU

1 Answers