small change in mpi GPU config

cvfinetune/finetuner/mpi.py

@@ -18,9 +18,10 @@ class MPIFinetuner(DefaultFinetuner):
 
 		self.comm = comm
 		if self.mpi:
-			self.device = opts.gpu[self.comm.rank]
-
-			# self.device += self.comm.intra_rank
+			if len(opts.gpu) > 1:
+				self.device = opts.gpu[self.comm.rank]
+			else:
+				self.device += self.comm.intra_rank
 
 	def scatter_datasets(self):
 		if self.mpi:

requirements.txt

@@ -13,4 +13,4 @@ cupy-cuda92~=4.2.0
 # my own packages
 cvargparse~=0.1.5
 cvdatasets~=0.4.2
-chainer_addons~=0.5.2
+chainer_addons~=0.6.0