GPU에 Tensorflow 작업이 적용되지 않았습니다.

Question

가능한 경우 모든 작업을 여기에 배치하려면 with tf.device('/gpu:0') 내부에 빌드 된 대형 모델이 있습니다. 그러나 1.4에서 1.2로 업데이트 한 후 tensorflow는 모든 작업을 CPU에 배치하려고 시도합니다 (log_device_placement에 따라). GPI에 배치 할 수없는 것처럼 보이는 몇 가지 연산 (AdamOptimizer뿐만 아니라 컴퓨팅 셰이프와 관련 있음)이 있기 때문에 소프트 장치 배치를 해제 할 수 없습니다. 거기에 몇 가지 중간 옵션을 나는 그것이 gpu 커널에없는 커널을 제외한 모든 작업을 배치 tensorflow에 줄 수 있습니까?

Answer 1

핍 포장 (https://github.com/tensorflow/tensorflow/issues/12388)의 버그로 인해 실제로 tensorflow-gpu가 실행되지 않는 것으로 보입니다. pip uninstall tensorflow && pip install tensorflow-gpu 이후 모든 것이 잘 작동하는 것 같습니다.