tensorflow가있는 keras에서 GPU를 비활성화하는 방법은 무엇입니까?

Question

내 코드의 처리 시간과 gpu의 처리 시간을 비교하고 싶습니다. 케라의 백엔드는 Tensorflow입니다. 그래서 GPU를 자동으로 사용합니다. 비교를 위해 keras/examples/mnist_mlp.py이라는 모델을 사용합니다.

다음과 같이 처리 시간을 확인했습니다. 그런 다음 GPU를 어떻게 비활성화합니까? ~/.keras/keras.json을 (를) 수정해야합니까?

$ time python mnist_mlp.py 
Test loss: 0.109761892007 
Test accuracy: 0.9832 
python mnist_mlp.py 38.22s user 3.18s system 162% cpu 25.543 total 

Answer 1

이런 식으로 시도한 적이 있습니까? :

$ CUDA_VISIBLE_DEVICES='' time python mnist_mlp.py 

CUDA_VISIBLE_DEVICES은 일반적으로 CUDA 일부 GPU의의를 숨기는 데 사용됩니다. 여기 보이는 장치를 넣지 않으면 숨길 수 있습니다.