Tensor flow toggle between CPU/GPU

前端 未结 6 1182
情歌与酒
情歌与酒 2021-01-30 13:30

Having installed tensorflow GPU (running on a measly NVIDIA GeForce 950), I would like to compare performance with the CPU.

I am running the tensorFlow MNIST tutorial c

6条回答
  •  灰色年华
    2021-01-30 14:12

    To make GPU invisible

    export CUDA_VISIBLE_DEVICES=""
    

    To return to normal

    unset CUDA_VISIBLE_DEVICES
    

提交回复
热议问题