배포시 GPU 수를 변경하십시오.

Question

2-GPU 시스템에서 미세 조정 된 Inception v3을 사용하고 있습니다. 가 지금은 1 개 GPU와 다른 시스템에서 훈련 모델을 실행하려고하지만이 같은 오류가 발생했습니다 :

'tower_1/기울기/tower_1/conv0/BatchNorm/순간 노드에 장치를 할당 할 수 없습니다/enough_statistics/Sub_grad/BroadcastGradientArgs ': 명시 적 장치 사양'/ device : GPU : 1 '을 충족시킬 수 없습니다. 해당 사양에 맞는 장치가이 프로세스에 등록되어 있지 않습니다. 사용 가능한 장치 :/직업 : 로컬 호스트/복제 : 0/작업 : 0/CPU : 0/직업 : 로컬 호스트/복제 : 0/작업 : 0/GPU : 0

그것은 모델이 2를 원하는 것 같다 - 훈련받은 것과 같은 GPU 환경. 이 모델을 1 GPU 만 사용하도록 변환 할 수 있습니까?

Answer 1

두 가지를 변경하여 효과가있었습니다.

1. 전원을 켜고 세션에서 allow_soft_placement 옵션에 :

   config = tf.ConfigProto(allow_soft_placement=True) 
   sess = tf.Session(config=config) 
   

2. 이름 바꾸기 모델 파일 이름 model.ckpt-50000.data-00000-of-00001에서 model.ckpt-50000에