2017-03-22 9 views
-3

모델이 GPU로 학습되고 결과가 검사 점 파일로 저장됩니다. 저장된 체크 포인트 파일은 cpu-tensorflow로 실행할 수 있습니까? 그렇지 않다면 cpu-tensorflow에서 모델을 실행하기 위해 저장된 체크 포인트 파일을 변환 할 수 있습니까?Tensorflow : GPU에서 교육 된 모델 (검사 점 파일)을 모델을 실행중인 CPU로 변환 할 수 있습니까?

+7

가능한 [gpu에서 숙련 된 모델을 추측을 위해 CPU에서 사용할 수 있습니까?] (http://stackoverflow.com/questions/40980035/can-a-model-trained-on-gpu-used- on-cpu-for-inference 및 그 반대로) – kaufmanu

답변

0

예! 정상적으로 할 수 있습니다!

예외는 with tf.device('gpu:0') 문입니다. 코드에 포함시키지 않으면 갈 수 있습니다!

행운을 빈다.

+1

코드에 장치 할당이있는 경우 remove_device 재 작성 규칙 (https://github.com/)을 사용하여 저장된 GraphDef .pb 파일에서 장치 할당을 제거 할 수 있습니다. tensorflow/tensorflow/tree/master/tensorflow/tools/graph_transforms/# remove_device –