diff --git a/pipeline.py b/pipeline.py
index e6283fd..dfa0331 100644
--- a/pipeline.py
+++ b/pipeline.py
@@ -150,7 +150,7 @@ class Components :
 		else:
 			_args['gpu']  = 0
 		_args['num_gpu'] 	= 1
-		
+		os.environ['CUDA_VISIBLE_DEVICES'] = str(args['gpu']) 
 		_args['no_value']= args['no_value']
 		# MAX_ROWS = args['max_rows']  	if 'max_rows' in args else 0
 		PART_SIZE = int(args['part_size']) if 'part_size' in args else 8