Advertisement
Guest User

Untitled

a guest
Jul 21st, 2019
61
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.41 KB | None | 0 0
  1. n_gpu = 4
  2. if(n_gpu == 1):
  3. self._build_model()
  4. self._build_train_op()
  5. else:
  6. # 1 GPU for training, n-1 for playing
  7. if(self.id <= 3):
  8. gpu_id = 0
  9. else:
  10. gpu_id = (1 + (self.id%(n_gpu-1)))
  11. os.environ["CUDA_VISIBLE_DEVICES"]=str(gpu_id)
  12. self._build_model()
  13. self._build_train_op()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement